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

Merge pull request #371 from chengzhengxin/master 

ctdet_decode support cat_spec_wh
This commit is contained in:
Xingyi Zhou 2019-09-20 15:42:32 -05:00 committed by GitHub
parent 9781e7defd 8d95cc380f
commit 9efed05d53
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/lib/detectors/ctdet.py
View File

@ -37,7 +37,7 @@ class CtdetDetector(BaseDetector):
reg = reg[0:1] if reg is not None else None reg = reg[0:1] if reg is not None else None
torch.cuda.synchronize() torch.cuda.synchronize()
forward_time = time.time() forward_time = time.time()
dets = ctdet_decode(hm, wh, reg=reg, K=self.opt.K) dets = ctdet_decode(hm, wh, reg=reg, cat_spec_wh=self.opt.cat_spec_wh, K=self.opt.K)
if return_time: if return_time:
return output, dets, forward_time return output, dets, forward_time
@ -93,4 +93,4 @@ class CtdetDetector(BaseDetector):
for bbox in results[j]: for bbox in results[j]:
if bbox[4] > self.opt.vis_thresh: if bbox[4] > self.opt.vis_thresh:
debugger.add_coco_bbox(bbox[:4], j - 1, bbox[4], img_id='ctdet') debugger.add_coco_bbox(bbox[:4], j - 1, bbox[4], img_id='ctdet')
debugger.show_all_imgs(pause=self.pause) debugger.show_all_imgs(pause=self.pause)