transform的使用1

最新推荐文章于 2024-11-16 20:22:36 发布

小驰驰呕吼**

最新推荐文章于 2024-11-16 20:22:36 发布

阅读量704

点赞数

分类专栏： pytorch 文章标签：神经网络深度学习机器学习

本文链接：https://blog.csdn.net/weixin_43328054/article/details/122929863

版权

pytorch 专栏收录该内容

20 篇文章 1 订阅

订阅专栏

from PIL import Image
from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter
import cv2

writer = SummaryWriter("logs")

# 1. transforms如何使用

img_path = "dataset/train/ants/0013035.jpg"
img = Image.open(img_path)
# 创建对象
tensor_trans = transforms.ToTensor()
img_tensor = tensor_trans(img)
print(img_tensor)

writer.add_image("Tensor_img", img_tensor)
writer.close()

# 2. 为什么使用Tensor数据类型
# 可以看成tensor包含了神经网络需要的一些参数