fix calass

7181010e · FlyingQianMM · 336a4fef · 7181010e
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddlex/cv/nets/detection/mask_rcnn.py paddlex/cv/nets/detection/mask_rcnn.py +1 -1

未找到文件。
--- a/paddlex/cv/nets/detection/mask_rcnn.py
+++ b/paddlex/cv/nets/detection/mask_rcnn.py
@@ -204,7 +204,7 @@ class MaskRCNN(object):
                bg_thresh_hi=self.bg_thresh_hi,
                bg_thresh_lo=self.bg_thresh_lo,
                bbox_reg_weights=self.bbox_reg_weights,
-                calass_nums=self.num_classes,
+                class_nums=self.num_classes,
                use_random=self.rpn_head.use_random)

            rois = outputs[0]