【PyTorch笔记】pytorch入门教程3transforms

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)这一句话