15 lines
539 B
Python
15 lines
539 B
Python
from PIL import Image
|
||
from torch.utils.tensorboard import SummaryWriter
|
||
from torchvision import transforms
|
||
|
||
img_path = r"C:\Users\12642\Desktop\Picture\dataset\train\ants\6240338_93729615ec.jpg"
|
||
img = Image.open(img_path)
|
||
|
||
writer = SummaryWriter("logs")
|
||
|
||
# transforms.ToTensor() 将 PIL.Image 或者 numpy.ndarray 转化为 tensor
|
||
tensor_trans = transforms.ToTensor()
|
||
tensor_img = transforms.ToTensor()(img)
|
||
|
||
writer.add_image("tensor_img", tensor_img) # 这里的tensor_img是一个三维的tensor，所以可以直接显示
|
||
writer.close() |