From 612acfa8acdc6022a64d223d703f8aa4a28a58b1 Mon Sep 17 00:00:00 2001 From: sunyanfang01 Date: Thu, 28 May 2020 14:19:13 +0800 Subject: [PATCH] modify transform --- paddlex/cv/models/yolo_v3.py | 4 ++++ paddlex/cv/transforms/det_transforms.py | 1 + 2 files changed, 5 insertions(+) diff --git a/paddlex/cv/models/yolo_v3.py b/paddlex/cv/models/yolo_v3.py index ed1f904..d19a489 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 3df697c..a72a920 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: 形参数据类型不满足需求。 -- GitLab