我是靠谱客的博主 呆萌书本,最近开发中收集的这篇文章主要介绍Yolov5系列(一)——网络可视化工具netron一、介绍二、安装步骤(介绍windows系统安装)三、打开yolov5系列的模型四、执行命令可能遇到的问题,觉得挺不错的,现在分享给大家,希望可以做个参考。
概述
一、介绍
netron:网络可视化工具,使用这个软件可以清晰的展示网络模型的每一层的输入输出,网络的整体的架构,而且支持不同的网络框架。
类似下图:
二、安装步骤(介绍windows系统安装)
(其他版本可以参考博客)
1、打开此链接:Github下载链接
打开上述链接,划到下面 install,选择 windows 下载
2、选择下图所示Netron-Setup-5.8.9.exe,下载安装
3、安装成功之后显示如下:
三、打开yolov5系列的模型
1、转换
因为netron对 .pt 的文件没有太好的兼容性,所以需要将文件进行转换,在yolov5 工程中,提供了export.py,可以进行文件转换
2、在终端直接输入以下命令
python export.py --weights yolov5s.pt --include onnx
注意:输入命令时需要注意export.py 和 yolo5s.pt 的路径
我是直接在该文件的目录下,所以不需要再写其他的路径
执行完命令,会在yolov5s.pt 同级目录下生成 yolov5s.onnx
此时就可以用网络可视化工具netron打开
四、执行命令可能遇到的问题
requirements: onnx not found and is required by YOLOv5, attempting auto-update...
requirements: 'pip install onnx' skipped (offline)
ONNX: export failure: No module named 'onnx'
解决办法
安装onnx
pip install onnx
之后再执行上述命令就没有问题了
最后
以上就是呆萌书本为你收集整理的Yolov5系列(一)——网络可视化工具netron一、介绍二、安装步骤(介绍windows系统安装)三、打开yolov5系列的模型四、执行命令可能遇到的问题的全部内容,希望文章能够帮你解决Yolov5系列(一)——网络可视化工具netron一、介绍二、安装步骤(介绍windows系统安装)三、打开yolov5系列的模型四、执行命令可能遇到的问题所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复