#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
@Time        : 2022/4/26 10:06
@Author      : Albert Darren
@Contact     : 2563491540@qq.com
@File        : TensorBoard_learn_2.py
@Version     : Version 1.0.0
@Description : TODO
@Created By  : PyCharm
"""
from torch.utils.tensorboard import SummaryWriter
from PIL import Image
import numpy as np

img_path = "Data/train/ants_image/5650366_e22b7e1065.jpg"
img_pillow = Image.open(img_path)
img_arr = np.array(img_pillow)
# tensorboard --logdir=logs --host=127.0.0.1
writer = SummaryWriter(log_dir="logs")
writer.add_image(tag="ants0", img_tensor=img_arr, global_step=1, dataformats="HWC")
writer.close()