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

2
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