From 8eee60e2e17a2deec4f323d213c1ddece9b09585 Mon Sep 17 00:00:00 2001
From: sunyanfang01 <sunyanfang01@baidu.com>
Date: Sun, 31 May 2020 16:16:31 +0800
Subject: [PATCH] fix the max_shape

---
 paddlex/cv/models/yolo_v3.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlex/cv/models/yolo_v3.py b/paddlex/cv/models/yolo_v3.py
index 89c3bbc..b87b667 100644
--- a/paddlex/cv/models/yolo_v3.py
+++ b/paddlex/cv/models/yolo_v3.py
@@ -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,
-- 
GitLab