#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
@Time        : 2022/4/26 13:51
@Author      : Albert Darren
@Contact     : 2563491540@qq.com
@File        : _transforms.py
@Version     : Version 1.0.0
@Description : TODO
@Created By  : PyCharm
"""
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms

from PIL import Image

img_path = "Data/train/ants_image/0013035.jpg"
img = Image.open(img_path)
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)
writer = SummaryWriter("logs")
writer.add_image(tag="ants1", img_tensor=tensor_img)
writer.close()