小土堆Pytorch笔记P8、9

编程基础很弱，需要机器学习，学习记录，按自己理解写的，希望以后能学懂吧，要是有大神看到还请赐教。

TensorBoard的使用（一）

关注于类的使用

Pycharm看帮助按住Ctrl左键想查看的

按照up视频运行一下代码报错。

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")

for i in range(100):
    writer.add_scalar(tag="y=x",scalar_value=i,global_step=i)

报错内容：

 AttributeError: module 'distutils' has no attribute 'version'

AttributeError: module ‘distutils‘ has no attribute ‘version‘ 解决方案 - 知乎 (zhihu.com)

根据此回答先

pip uninstall setuptools

然后

pip install setuptools==58.0.4

然后就可以运行了。

查看时间文件两种方法：

1.Pycharm的Terminal

在其中输入：(注意，这里要输入绝对路径)

tensorboard --logdir="D:\\Scientific_Research\\Pycharmproject\\Biji\\logs"

点击后显示为：

画y=2x

from torch.utils.tensorboard import SummaryWriter

# 创建类示例
writer = SummaryWriter("logs")
# 画 y=2x
for i in range(100):
    writer.add_scalar(tag="y=2x", scalar_value=2 * i, global_step=i)

TensorBoard的使用（二）

add_image()的使用（常用来观察训练结果）

视频中赋予相对地址，我没弄明白，这里用绝对地址凑活吧。

add_image()中img_tensor需要的数据类型为torch.tensor或numpy.array，使用PIL读取图片类型不符合。

两种方法搞定类型问题：

1.np.array()将PIL打开的图片转化为对应类型

img_array=np.array(img)

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

writer = SummaryWriter("logs")
image_path="D:\\Scientific_Research\\Pycharmproject\\Biji\\data\\train\\ants_image\\0013035.jpg"#定义地址
img_PIL=Image.open(image_path)#PIL打开图片
img_array=np.array(img_PIL)

#ndarray图片shape为（高，宽，通道数）
#在输入时需要指定
writer.add_image("名称",img_array, 1 ,dataformats='HWC')

2.使用OpenCv