yolov5模型代码怎么修改_yolov5源码解析修改

作者：羊村懒王 | 2024-04-20 10:49:07

踩

yolov5源码解析修改

yaml配置文件

深度乘积因子
宽度乘积因子
所有版本只有这两个参数的不同，s m l x逐渐加宽加深

在这里插入图片描述

各种类型层参数对照

在这里插入图片描述
backbone里的各层，在这里解析，只需要改.yaml里的各层参数就能控制网络结构

修改网络结构

第一步：把新加的模块层加到experimental.py文件里
在这里插入图片描述
记得修改完，记得把新定义的module类写到yolo.py文件里，就是在这些模块里…然后加输入通道数那块
就这个函数里，这些逻辑

新加的话，要把新加的模块的参数计算逻辑写在elif里
然后就放进nn.sequential里了
在这里插入图片描述
yolo文件就是整个模型结构的定义nnsequential
修改完先跑一下yolo.py看看能不能输出正确的网络结构

第二步：看它的参数，在配置文件里写对应的参数
在这里插入图片描述

增加检测层

在这里插入图片描述

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