import torchvision.transforms as transforms
from tensorboardX import SummaryWriter
from PIL import Image
writer = SummaryWriter('logs')
img_path = r'D:\360Downloads\pytorch\data\train\ants_image\7759525_1363d24e88.jpg'
from PIL import Image
img = Image.open(img_path)
tensor_trans = transforms.ToTensor()
#引用ToTensor这个class'
tensor_img = tensor_trans(img)
writer.add_image('Tensor_img',tensor_img)
#writer.add_image参数1是显示的名称,参数2是一个tensor类型
writer.close()
#终端输入:tensorboard --logdir=logs --port=9999
import os
root_dir = 'data/train'
target_dir='ants_image'
img_path = os.listdir(os.path.join(root_dir,target_dir))
label=target_dir.split('_')[0]
out_dir ='ants_label'
for i in img_path:
file_name = i.split('.jpg')[0]
with open(os.path.join(root_dir,out_dir,'{}.txt'.format(file_name)),'w') as f:
f.write(label)
#创建图片名的txt在out_dir路径下,名称是'{}.txt'.format(file_name)。内容是ants,因为有f.write(label)这一句话
【PyTorch笔记】pytorch入门教程3transforms
于 2023-05-13 16:26:35 首次发布