towhee
/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
22 lines
768 B
22 lines
768 B
from torchvision import transforms
|
|
from transform_image_operator_template import TransformImageOperatorTemplate
|
|
|
|
|
|
def get_transformed_img(size, img_path)
|
|
tfms = transforms.Compose(
|
|
[
|
|
transforms.Resize(size),
|
|
transforms.CenterCrop(224),
|
|
transforms.ToTensor(),
|
|
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),
|
|
]
|
|
)
|
|
img_transformed = tfms(img_path).unsqueeze(0)
|
|
return img_transformed
|
|
|
|
def test_transform_image(size=256, img_path='./test_data/test.jpg'):
|
|
op = TransformImageOperatorTemplate(size
|
|
outputs = op(img_path)
|
|
img_transformed = get_transformed_img(size, img_path)
|
|
c = (img_transformed.numpy() == outputs.img_transformed.numpy())
|
|
assert c.all()
|