YOLOv5更换Ghostnet模块

在最新版本的YOLOv5-6.1源码中,作者已经加入了Ghost模块,在models/common.py文件下

YOLOv5更换Ghostnet模块_第1张图片

并在models/hub/文件夹下,给出了yolov5s-ghost.yaml文件,因此直接使用即可

YOLOv5更换Ghostnet模块_第2张图片

在用的时候注意train.py里面的'--cfg'配置文件

YOLOv5更换Ghostnet模块_第3张图片

 更改完毕,直接运行就可以啦~~~

reference:【魔改YOLOv5-6.x(上)】结合轻量化网络Shufflenetv2、Mobilenetv3和Ghostnet_嗜睡的篠龙的博客-CSDN博客_yolov5魔改

你可能感兴趣的:(python,开发语言)