diff --git a/mmdet/models/detectors/htc.py b/mmdet/models/detectors/htc.py index 6f8eef50146730d9e9d89c9ba99c60c8fd39abad..097d109ac488aefa71408c63c468f4163361fc1a 100644 --- a/mmdet/models/detectors/htc.py +++ b/mmdet/models/detectors/htc.py @@ -342,7 +342,7 @@ class HybridTaskCascade(CascadeRCNN): img_shape, scale_factor, rescale=rescale, - nms_cfg=rcnn_test_cfg) + cfg=rcnn_test_cfg) bbox_result = bbox2result(det_bboxes, det_labels, bbox_head.num_classes) ms_bbox_result['stage{}'.format(i)] = bbox_result