pytorch的tensor，PIL和numpy的相互转换

最新推荐文章于 2024-08-15 19:57:00 发布

峰峰的猫

最新推荐文章于 2024-08-15 19:57:00 发布

阅读量3.3k

点赞数

分类专栏：图像处理相关工具docker+git

本文链接：https://blog.csdn.net/weixin_42662358/article/details/89761314

版权

图像处理同时被 2 个专栏收录

11 篇文章 0 订阅

订阅专栏

相关工具docker+git

4 篇文章 0 订阅

订阅专栏

原图：
在这里插入图片描述

# coding: utf-8
"""
create by ljf on 2019/3/8 15:02
"""
import torch
from torchvision import transforms
from PIL import Image

#1.PIL转为tensor
transform = transforms.ToTensor()
img = Image.open("scene.jpg")
tensor = transform(img)
# print(tensor.size())

#2.对图像增加随机翻转，随机剪切操作
transform = transforms.Compose([transforms.RandomCrop((224,224),padding=4),
                                transforms.RandomHorizontalFlip()])
img2 = transform(img)
# img2.show()
img2.save("scene-crop.jpg")

#3.tensor转换为numpy
arr = tensor.cpu().data.numpy()
print(arr.shape)