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

modified ctdet_decode to support cat_spec_wh

This commit is contained in:
chengzhengxin 2019-09-20 14:58:12 +08:00
parent 9781e7defd
commit 8d95cc380f
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