Tensor数据类型

opencv+totensor+tensorboard 使用

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

writer = SummaryWriter('logs')
tool = transforms.ToTensor()

img_path = r'data/hymenoptera_data/hymenoptera_data/train/ants/9715481_b3cb4114ff.jpg'
img_array = cv2.imread(img_path)  # 返回array

img_tensor = tool(img_array)
writer.add_image('train', img_tensor, 1)
writer.close()

标签：Tensor,img,writer,数据类型,transforms,import,array,data
来源： https://blog.csdn.net/qq_49821869/article/details/122764457