opencv+totensor+tensorboard 使用
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
import cv2
writer = SummaryWriter('logs')
tool = transforms.ToTensor()
img_path = r'data/hymenoptera_data/hymenoptera_data/train/ants/9715481_b3cb4114ff.jpg'
img_array = cv2.imread(img_path) # 返回array
img_tensor = tool(img_array)
writer.add_image('train', img_tensor, 1)
writer.close()
