Merge pull request #371 from chengzhengxin/master

ctdet_decode support cat_spec_wh
2019-09-20 15:42:32 -05:00 · 2019-09-20 15:42:32 -05:00 · 9efed05d53
parent 9781e7defd 8d95cc380f
commit 9efed05d53
1 changed files with 2 additions and 2 deletions
--- a/src/lib/detectors/ctdet.py
+++ b/src/lib/detectors/ctdet.py
@ -37,7 +37,7 @@ class CtdetDetector(BaseDetector):
        reg = reg[0:1] if reg is not None else None
      torch.cuda.synchronize()
      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:
      return output, dets, forward_time