Browse Source

back maskrcnn

FlyingQianMM 5 năm trước cách đây
mục cha
commit
66a50bcffd
1 tập tin đã thay đổi với 2 bổ sung1 xóa
  1. 2 1
      paddlex/cv/nets/detection/mask_rcnn.py

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

@@ -204,7 +204,8 @@ class MaskRCNN(object):
                 bg_thresh_hi=self.bg_thresh_hi,
                 bg_thresh_lo=self.bg_thresh_lo,
                 bbox_reg_weights=self.bbox_reg_weights,
-                class_nums=self.num_classes)
+                calass_nums=self.num_classes,
+                use_random=self.rpn_head.use_random)
 
             rois = outputs[0]
             labels_int32 = outputs[1]