我是靠谱客的博主 呆萌书本,最近开发中收集的这篇文章主要介绍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系列的模型四、执行命令可能遇到的问题所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(56)

评论列表共有 0 条评论

立即
投稿
返回
顶部