YOLOv5转onnx格式文件,并将其用netron可视化 |
您所在的位置:网站首页 › h5转onnx › YOLOv5转onnx格式文件,并将其用netron可视化 |
YOLOv5转onnx格式文件,并将其用netron可视化
git clone YOLOv5
转换成onnx文件格式。
使用netron工具来可视化
git clone YOLOv5
首先,我们需要进入github官网,找到yolov5的源代码,这里附上链接YOLOv5 点击链接后即可进入Glenn大神写的YOLO系列代码库中(用内网的话有点慢,有可能需要多刷新几次才能出来,耐心等待一下),顺利的话我们可以看到: 本篇博客讲的是YOLOv5的网络结构可视化,因此我们可以在左上角选择版本,这里链接进入的默认是选择的v5.0。如果你想玩一玩其他的版本,也可以点击Glenn大神头像上方的v5.0来进行一个版本的选择。点击之后会看到下图,选择Tags就可以查看各种版本了。这里我们以5.0版本为例。 然后就可以在本地白嫖到这个YOLOv5的项目。 由于我是习惯用pycharm来开发,此时打开用pycharm以项目的方式打开YOLOv5,点击下方的Terminal栏目,进入项目的终端命令行中: 可以下载该项目所依赖的各种包,当然这里其实并不齐全,还有一些包在运行的时候会报错,这个时候我们就手动的pip install一下就可以了,不会消耗很多的时间。 到这里我们就完成了第一步,将YOLOv5项目部署在我们的电脑上了。 转换成onnx文件格式。进入这一步后,首先我们要知道的是,对于netron工具来说,要想可视化我们的网络架构,需要获得一个onnx格式的文件,而这个文件可以通过我们的pt文件转换而来,也就是我们的权重文件。这里我们可以先进入项目中的model文件下的export.py文件中,运行之后我们会得到一个yolov5s.pt这样一个预训练的权重文件(当然这只是举个例子,对于我们自己训练得到的权重文件也是一样的操作)。这里我们在下载好这个预训练权重文件后,一般来说代码会报错,这里我们先不管它,得到这个pt文件即可。随后我们在我们的pycharm终端中,或者git工具中,我这里打开的是git工具,进行onnx格式的转换。
然后回车运行: 然后就进入我们的最后一步,可视化。在浏览器上直接搜索(或者也可以进入github中,下载netron工具,具体的下载方式网上有讲解,这里我就不具体展开了),我这里用浏览器在线转换的方式呈现: 打开必应浏览器,输入netron,首页就出现了netron的官网: |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |