提交 66a50bcf 编写于 作者: F FlyingQianMM

back maskrcnn

上级 2c8d51cb
......@@ -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]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册