opencv dnn部署yolov5的口罩检测模型+GUI界面(三)_dnn口罩模型

作者：Gausst松鼠会 | 2024-03-20 15:34:34

踩

dnn口罩模型

我训练完自己的口罩检测模型，得到了一个best.pt文件，在detect.py里运行无误后，尝试把模型导出。

yolov5官方代码的训练出来模型的权重是保存在best.pt文件里的。这个.pt后缀的模型文件并不常规，如果想要脱离训练环境部署模型，需要对原始的.pt文件进行导出操作，这里我把模型导出为onnx文件，想要用opencv的dnn模块读取该onnx模型文件进行推理，这样操作之后，整个检测过程就只需要import cv2这一个第三方库了。

yolov5官方代码已经提供了export.py函数，修改一下运行参数就支持导出为onnx。

进入这个网站Netron 查看一下网络结构。

在里面选择自己的best.pt，这里以我的口罩检测模型为例，看到网络结构如下

查看.pt文件时，每一层网络并未做展开，看着很难受。

下面执行export.py将自己的模型文件导出为onnx文件。

吹爆这位大佬的博客：

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/274670