跟着小土堆学pytorch（p9.Transforms的使用（二））

2.为什么使用Tensor数据类型

因为Tensor数据类型是复合结构，其中有用于计算梯度的参数

1.将代码放入pythong console中

 

Tensor：满足神经网络学习需要的参数的一种数据类型 

使用opencv中的cv2.imread()读取ndarray格式图片

import cv2
cv_img = cv2.imread(img_path)

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

""""python的用法 ——》Tensor数据类型
    通过transform.ToTensor去解决两个问题
    1.transform如何使用
    2.为什么需要Tensor数据类型
"""

img_path = "dataset/train/bees/2822388965_f6dca2a275.jpg"
img = Image.open(img_path)
print(img)

writer = SummaryWriter("logs")

tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)

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

(mypytorch) C:\Users\Administrator\PycharmProjects\learn_pytorch>tensorboard --logdir=logs 
TensorFlow installation not found - running with reduced feature set.
Serving TensorBoard on localhost; to expose to the network, use a proxy or pass --bind_all
TensorBoard 2.12.0 at http://localhost:6006/ (Press CTRL+C to quit)