yolov5模型代码怎么修改
yaml配置文件
深度乘积因子
宽度乘积因子
所有版本只有这两个参数的不同,s m l x逐渐加宽加深
各种类型层参数对照
backbone里的各层,在这里解析,只需要改.yaml里的各层参数就能控制网络结构
修改网络结构
第一步:把新加的模块层加到experimental.py文件里
记得修改完,记得把新定义的module类写到yolo.py文件里,就是在这些模块里…然后加输入通道数那块
就这个函数里,这些逻辑
新加的话,要把新加的模块的参数计算逻辑写在elif里
然后就放进nn.sequential里了
yolo文件就是整个模型结构的定义nnsequential
修改完先跑一下yolo.py看看能不能输出正确的网络结构
第二步:看它的参数,在配置文件里写对应的参数