提交 8eee60e2 编写于 作者: S sunyanfang01

fix the max_shape

上级 279347c0
......@@ -266,7 +266,7 @@ class YOLOv3(BaseAPI):
if isinstance(bt, paddlex.det.transforms.BatchRandomShape):
self.max_shape = max(bt.random_shapes)
break
self.init_params['max_shape'] = max_shape
self.init_params['max_shape'] = self.max_shape
iou_bt = paddlex.det.transforms.GenerateYoloTarget
train_dataset.transforms.batch_transforms.append(iou_bt(anchors=self.anchors,
anchor_masks=self.anchor_masks,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册