From 8d95cc380f68d1a287039ed0a37b93be8fd763a3 Mon Sep 17 00:00:00 2001 From: chengzhengxin Date: Fri, 20 Sep 2019 14:58:12 +0800 Subject: [PATCH] modified ctdet_decode to support cat_spec_wh --- src/lib/detectors/ctdet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)