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()
|