diff --git a/paddlex/cv/models/yolo_v3.py b/paddlex/cv/models/yolo_v3.py index ed1f904606a79d1ecfb2fe98ae6f4eca40faed11..d19a4894c17b51f82f550595441f590a17ec7e27 100644 --- a/paddlex/cv/models/yolo_v3.py +++ b/paddlex/cv/models/yolo_v3.py @@ -249,6 +249,10 @@ class YOLOv3(BaseAPI): self.train_batch_size = train_batch_size self.labels = train_dataset.labels + if pretrain_weights == "Object365": + for transform in train_dataset.transforms.transforms: + if isinstance(transform, paddlex.det.transforms.Normalize): + transform.is_scale = False if self.use_iou_loss or self.use_iou_aware_loss: if self.train_random_shapes is None or len(self.train_random_shapes) == 0: for transform in train_dataset.transforms.transforms: diff --git a/paddlex/cv/transforms/det_transforms.py b/paddlex/cv/transforms/det_transforms.py index 3df697c249448e716c4e7d2350a3d87fd7cf4a7e..a72a9205e77edd1b1ff0c1f8533f98d5fc00e77a 100644 --- a/paddlex/cv/transforms/det_transforms.py +++ b/paddlex/cv/transforms/det_transforms.py @@ -494,6 +494,7 @@ class Normalize(DetTransform): Args: mean (list): 图像数据集的均值。默认为[0.485, 0.456, 0.406]。 std (list): 图像数据集的标准差。默认为[0.229, 0.224, 0.225]。 + is_scale (bool): 是否对图像归一化。默认为True。 Raises: TypeError: 形参数据类型不满足需求。