From fa2db1599019b5ef862b0845e91b5460e3f1dcaa Mon Sep 17 00:00:00 2001
From: Trayvon <31656547+WangChuanYuan@users.noreply.github.com>
Date: Wed, 23 Oct 2019 18:32:02 +0800
Subject: [PATCH] Fix wrong keyword argument 'nms_cfg' in htc (#1573)
---
mmdet/models/detectors/htc.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mmdet/models/detectors/htc.py b/mmdet/models/detectors/htc.py
index 6f8eef5..097d109 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
--
GitLab