diff --git a/src/lib/detectors/ctdet.py b/src/lib/detectors/ctdet.py index 4f97624..5bf0594 100755 --- 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 @@ -93,4 +93,4 @@ class CtdetDetector(BaseDetector): for bbox in results[j]: if bbox[4] > self.opt.vis_thresh: debugger.add_coco_bbox(bbox[:4], j - 1, bbox[4], img_id='ctdet') - debugger.show_all_imgs(pause=self.pause) \ No newline at end of file + debugger.show_all_imgs(pause=self.pause)