YOLOv5更换Ghostnet模块
在最新版本的YOLOv5-6.1源码中,作者已经加入了Ghost模块,在models/common.py文件下并在models/hub/文件夹下,给出了yolov5s-ghost.yaml文件,因此直接使用即可在用的时候注意train.py里面的'--cfg'配置文件更改完毕,直接运行就可以啦~~~reference:【魔改YOLOv5-6.x(上)】结合轻量化网络Shufflenetv2、Mob
·
在最新版本的YOLOv5-6.1源码中,作者已经加入了Ghost模块,在models/common.py文件下
并在models/hub/
文件夹下,给出了yolov5s-ghost.yaml
文件,因此直接使用即可
在用的时候注意train.py里面的'--cfg'配置文件
更改完毕,直接运行就可以啦~~~
reference:【魔改YOLOv5-6.x(上)】结合轻量化网络Shufflenetv2、Mobilenetv3和Ghostnet_嗜睡的篠龙的博客-CSDN博客_yolov5魔改
更多推荐
已为社区贡献5条内容
所有评论(0)