Skip to content

GitLab

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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
mmdet/models/detectors/htc.py
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
......
Markdown is supported
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