From f001960b7359f3a88b7dd96e1f34500b90566ceb Mon Sep 17 00:00:00 2001 From: jiangjiajun Date: Wed, 27 May 2020 21:18:41 +0800 Subject: [PATCH] fix key error --- paddlex/cv/nets/detection/mask_rcnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddlex/cv/nets/detection/mask_rcnn.py b/paddlex/cv/nets/detection/mask_rcnn.py index a5dd018..b67f44a 100644 --- 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] -- GitLab