OrgLion-ML
/
CenterNet
mirror of https://github.com/xingyizhou/CenterNet.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Merge pull request #17 from cclauss/patch-1 

Add comments to explain why nms = None
This commit is contained in:
Xingyi Zhou 2019-04-22 20:37:03 -05:00 committed by GitHub
parent c4792e3955 1316d370fe
commit cefeb33baf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tools/voc_eval_lib/model/test.py
View File

@ -21,7 +21,8 @@ from utils.blob import im_list_to_blob
from model.config import cfg, get_output_dir from model.config import cfg, get_output_dir
from model.bbox_transform import clip_boxes, bbox_transform_inv from model.bbox_transform import clip_boxes, bbox_transform_inv
# from model.nms_wrapper import nms # from model.nms_wrapper import nms # need to compile cython nms before import nms
nms = None # not needed in pascal evaluation
def _get_image_blob(im): def _get_image_blob(im):
"""Converts an image into a network input. """Converts an image into a network input.
@ -190,4 +191,3 @@ def test_net(sess, net, imdb, weights_filename, max_per_image=100, thresh=0.):
print('Evaluating detections') print('Evaluating detections')
imdb.evaluate_detections(all_boxes, output_dir) imdb.evaluate_detections(all_boxes, output_dir)