Merge pull request #262 from hellock/hotfix

bug fix for argument name

Merge pull request #262 from hellock/hotfix
bug fix for argument name
e72a9fd5 · Kai Chen · GitHub · 432c3320 · d49b7f7f · e72a9fd5
Unverified Commit e72a9fd5 authored Jan 16, 2019 by Kai Chen Committed by GitHub Jan 16, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet/models/anchor_heads/ssd_head.py mmdet/models/anchor_heads/ssd_head.py +1 -1

No files found.
--- a/mmdet/models/anchor_heads/ssd_head.py
+++ b/mmdet/models/anchor_heads/ssd_head.py
@@ -144,7 +144,7 @@ class SSDHead(AnchorHead):
            self.target_stds,
            cfg,
            gt_labels_list=gt_labels,
-            cls_out_channels=self.cls_out_channels,
+            label_channels=1,
            sampling=False,
            unmap_outputs=False)
        if cls_reg_targets is None: