from torch.utils.tensorboard import SummaryWriter
import numpy as np
from PIL import Image

writer = SummaryWriter('logs')
image_path = "dataset/train/ants/5650366_e22b7e1065.jpg"
img_PIL = Image.open(image_path)
img_arr = np.array(img_PIL)


writer.add_image("test", img_arr, 2, dataformats='HWC')
# for i in range(100):
#     writer.add_scalar("y=2x", 2*i, i)




writer.close()