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


witer = SummaryWriter("logs")
img_path = "data/train/bees_image/16838648_415acd9e3f.jpg"
img_PIL = Image.open(img_path)
img_array = np.array(img_PIL)
print(type(img_array))
print(img_array.shape)


witer.add_image("train", img_array, 1, dataformats='HWC')
# y = x
for i  in range(100):
    witer.add_scalar("y = 2x", 3*i, i)


witer.close()