Skip to content
Snippets Groups Projects
Commit c992f24e authored by Cathy Wong's avatar Cathy Wong
Browse files

[MD] Transforms Unification - Minor Updates#2

Update due to PR2528 new model DnCNN - research/cv/DnCNN/infer/data/preprocess/export_bin_file.py
parent bd2c1a6f
Branches
Tags
No related merge requests found
......@@ -23,7 +23,7 @@ import numpy as np
import cv2
import mindspore
import mindspore.dataset as ds
import mindspore.dataset.transforms.c_transforms as C
import mindspore.dataset.transforms as C
def ResziePadding(img, fixed_side=256):
......
......@@ -31,7 +31,7 @@ class SingleScaleTrans:
def __call__(self, imgs, ann, image_names, image_size, batch_info):
size = self.resize
decode = P.Decode(True)
decode = V.Decode(True)
resize_letter_box_op = ResizeLetterbox(input_dim=size)
to_tensor = V.ToTensor()
......@@ -204,11 +204,11 @@ def preprocess_fn(image, annotation):
anchors = config.anchors
anchors_mask = config.anchors_mask
decode = P.Decode(True)
decode = V.Decode(True)
random_crop_letter_box_op = RandomCropLetterbox(jitter=jitter, input_dim=size)
random_flip_op = RandomFlip(flip)
hsv_shift_op = HSVShift(hue, sat, val)
to_tensor = P.ToTensor()
to_tensor = V.ToTensor()
img_pil = decode(image)
input_data = img_pil, annotation
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment