TensorBoardX add_image()输出图片与torchvision.transforms.Normalize()标准化
版权归属:
更多关注:
1、TensorBoardX add_image()输出图片需要注意的地方
pytorch中调用TensorBoardX显示图片的方法如下:
from tensorboardX import SummaryWriter
writer = SummaryWriter(log_dir)
# display image
writer.add_image('label_name', img, global_step=total_step)
首先,label_name必须是一个唯一的名字,特别需要注意的是防止与add_scale()中的名字重复,这样子会出现莫名的错误;
其次,参数img可以是numpy or torch.Tenso

本文介绍了在使用TensorBoardX的add_image()输出图片时应注意的事项,包括label_name的唯一性、图像维度和值域范围。同时,详细讨论了torchvision.transforms.Normalize()在图像标准化过程中的作用,指出其会将图像值域转换到[-1, 1],并提供了相关转换的公式。"
127674151,8086594,YOLOv5改进:添加SE注意力机制,"['目标检测', '深度学习', '计算机视觉', '神经网络', 'YOLO']
最低0.47元/天 解锁文章
1万+





