人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)

transform 主要是对input图像进行变换(统一尺寸、对图像中的数据进行类的转换)

TensorBoard很有用
如:通过loss的变化过程,来看loss的变化是否复合预想。也可以通过loss来选择模型。

人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第1张图片

TensorBoard,虽然他是TensorFlow 的一部分,但是可以独立安装,并且服务于Pytorch等其他的框架。

安装tensorboard
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第2张图片
在这里插入图片描述
但是这里tensorboard版本太低了,以下方法解决↓
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第3张图片
在pip list查看
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第4张图片

  • 事件文件
    SummaryWriter类,直接向一个文件夹里写入事件文件的一个类,可以被tensorboard解析
    人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第5张图片
    人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第6张图片

将tensorboard的事件文件放在logs文件夹里面

add_scalar()

人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第7张图片

  • 打开事件文件
    人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第8张图片
    如果一台服务器多个人训练,默认都打开这个端口的话会很不方便,所以可以去指定端口。
    可以通过设置参数去修改端口
    在这里插入图片描述
    点击端口即可
    人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第9张图片
    后续对代码有修改后,重新启动项目,在浏览器刷新即可

如果部分修改部分没修改,则会出现问题
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第10张图片
则会出现问题,把新的事件记录在了上个事件里
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第11张图片
可以把logs文件夹里的事件全删掉,重新启动程序、输入服务器

add_image()

人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第12张图片

↓这样的类型不符合要求
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第13张图片

  • 利用opencv读取数据,获得numpy数据
    人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第14张图片
    pip install opencv-python -i https://pypi.douban.com/simple/

人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第15张图片
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第16张图片
再进到端口中可以看到图片
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第17张图片
此时把代码中的图片换成蜜蜂,图片标题不变
则可通过拖动这个步骤条来看1、2步骤的图片(蚂蚁、蜜蜂)
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第18张图片
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第19张图片
要单独显示的话,只需要把这个title重新命名即可
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第20张图片
人工智能学习07--pytorch03--tensorboard(下载tensorboard、opencv)_第21张图片
通过这个方法,可以直观地显示训练中给model提供了哪些数据
或者相对model进行测试时,可以看到每个阶段的输出成果

你可能感兴趣的:(pytorch,人工智能,opencv,学习)