diff --git a/src/lib/detectors/exdet.py b/src/lib/detectors/exdet.py index 0bedac9..bcfae42 100755 --- a/src/lib/detectors/exdet.py +++ b/src/lib/detectors/exdet.py @@ -46,7 +46,7 @@ class ExdetDetector(BaseDetector): center_thresh=self.opt.center_thresh, aggr_weight=self.opt.aggr_weight) else: - dets = self.decode(t_heat, l_heat, b_heat, r_heat, c_heat, K=opt.K, + dets = self.decode(t_heat, l_heat, b_heat, r_heat, c_heat, K=self.opt.K, scores_thresh=self.opt.scores_thresh, center_thresh=self.opt.center_thresh, aggr_weight=self.opt.aggr_weight) @@ -129,4 +129,4 @@ class ExdetDetector(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='exdet') - debugger.show_all_imgs(pause=self.pause) \ No newline at end of file + debugger.show_all_imgs(pause=self.pause)