from torch.utils.tensorboard import SummaryWriter
导入(从torch.常用的工具箱中导入tensorboard,导入SummaryWriter类)
pycharm中按住Ctrl可查看属性等
一、TensorBoard的安装
https://blog.csdn.net/JineD/article/details/124774570
报错后用了上述方法三,又出现了新错误AttributeError: module ‘distutils‘ has no attribute ‘version‘
然后更新torch版本降低tensorboard的版本之后就安装成功了
二、add_scalar()的使用
add_scalar(“函数”,y轴数值,X轴数值)
logdir=事件文件所在的文件夹名
避免和别人端口一样
三、add_image()的使用
从PIL到numpy,需要在add_image()中指定shape中每一个数字/维表示的含义
eg: writer.add_image("test",img_array ,1,dataformats='HWC')
writer.add_image("test",img_array ,2,dataformats='HWC')
1、利用OpenCV读取图片,获得numpy型图片数据
在terminal安装OpenCV
报错原因是使用pip install opencv-python
命令安装的是最新版本,python3.6不支持。所以找一个python3.6支持的版本。如opencv-python==4.3.0.38
。
2、利用numpy.array(),对PIL图片进行转换