yolov5github原版设计网络使用的配置文件形式,
各种参数,跨文件配合使用,读懂整个网络不是特别方便:
如果想自己设计backbone,也不很方便,所以调研了一下,
寻找到参照主流的 pytorch 网络构建形式, 分成 backbone, neck, head 进行插件式构建, 实现支持 resnet, mobilenet, shufflenet,当然也可以自定义修改层,添加各种网络插件进行魔改。
本人整理的,带训练,预测,预训练模型,下载链接:
https://download.csdn.net/download/jacke121/21406799
问题:
1,34,2
5,1,2,
5,34
102,2
offsets = (torch.zeros_like(gxy)[None] + off[:, None])[j