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

fa2db159 · Trayvon · Kai Chen · 618dca08 · fa2db159
Commit fa2db159 authored Oct 23, 2019 by Trayvon Committed by Kai Chen Oct 23, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet/models/detectors/htc.py mmdet/models/detectors/htc.py +1 -1

No files found.
--- 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