OrgLion-ML
/
CenterNet
mirror of https://github.com/xingyizhou/CenterNet.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Undefined name: opt.K --> self.opt.K

This commit is contained in:
cclauss 2019-04-18 10:13:10 +02:00 committed by GitHub
parent d133939e9e
commit d0067e886d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/lib/detectors/exdet.py
View File

@ -46,7 +46,7 @@ class ExdetDetector(BaseDetector):
center_thresh=self.opt.center_thresh, center_thresh=self.opt.center_thresh,
aggr_weight=self.opt.aggr_weight) aggr_weight=self.opt.aggr_weight)
else: 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, scores_thresh=self.opt.scores_thresh,
center_thresh=self.opt.center_thresh, center_thresh=self.opt.center_thresh,
aggr_weight=self.opt.aggr_weight) aggr_weight=self.opt.aggr_weight)
@ -129,4 +129,4 @@ class ExdetDetector(BaseDetector):
for bbox in results[j]: for bbox in results[j]:
if bbox[4] > self.opt.vis_thresh: if bbox[4] > self.opt.vis_thresh:
debugger.add_coco_bbox(bbox[:4], j - 1, bbox[4], img_id='exdet') debugger.add_coco_bbox(bbox[:4], j - 1, bbox[4], img_id='exdet')
debugger.show_all_imgs(pause=self.pause) debugger.show_all_imgs(pause=self.pause)