Skip to content
Snippets Groups Projects
Commit fa2db159 authored by Trayvon's avatar Trayvon Committed by Kai Chen
Browse files

Fix wrong keyword argument 'nms_cfg' in htc (#1573)

parent 618dca08
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
mmdet/models/detectors/htc.py
+ 1
− 1
View file @ fa2db159
...@@ -342,7 +342,7 @@ class HybridTaskCascade(CascadeRCNN): ...@@ -342,7 +342,7 @@ class HybridTaskCascade(CascadeRCNN):
img_shape, img_shape,
scale_factor, scale_factor,
rescale=rescale, rescale=rescale,
nms_cfg=rcnn_test_cfg) cfg=rcnn_test_cfg)
bbox_result = bbox2result(det_bboxes, det_labels, bbox_result = bbox2result(det_bboxes, det_labels,
bbox_head.num_classes) bbox_head.num_classes)
ms_bbox_result['stage{}'.format(i)] = bbox_result ms_bbox_result['stage{}'.format(i)] = bbox_result
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment