Pytorch——SummaryWriter.add_image()的使用
from torch.utils.tensorboard import SummaryWriter用法image_path="data/train/ants_image/6240329_72c01e663e.jpg"#图片路径#读取图片的两种方式:img_PIL=Image.open(image_path) #方式一img_PIL=cv2.imread(image_path) #方式二img_array=np.array(img_PIL)writer=SummaryWriter("lo.
原创
2022-03-28 15:46:45 ·
5755 阅读 ·
0 评论