modified ctdet_decode to support cat_spec_wh

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