pytorch读取图片

最新推荐文章于 2024-08-24 11:47:11 发布

chenyu19880302

最新推荐文章于 2024-08-24 11:47:11 发布

阅读量4.3k

点赞数 2

分类专栏：基础知识文章标签： pytorch python

本文链接：https://blog.csdn.net/liulianfanjianshi/article/details/84935530

版权

基础知识专栏收录该内容

1 篇文章 0 订阅

订阅专栏

pytorch读取图片是用的python，目前刚开始学习，具体代码如下：

from torchvision import transforms, datasets as ds
from PIL import Image

img_path="F:/mpii_human_pose_v1/images/000001163.jpg"

transform1 = transforms.Compose([
    transforms.ToTensor(), # range [0, 255] -> [0.0,1.0]
    ]
)

##PIL
img = Image.open(img_path).convert('RGB') # 读取图像
img2 = transform1(img) # 归一化到 [0.0,1.0]
print("img2 = ",img2)
#转化为PILImage并显示
img_2 = transforms.ToPILImage()(img2).convert('RGB')
print("img_2 = ",img_2)
img_2.show()

正常显示