flexible-yolov5
基于 。
原始的Yolo V5是一个了不起的项目。 对于专业人士而言,理解和修改其代码应该不难。 我不是专家。 当我想对网络进行一些更改时,这并不是一件容易的事,例如添加分支和尝试其他骨干网。 也许会有像我这样的人,所以我将yolov5模型拆分为{主干,脖子,头部},以方便各种模块的操作并支持更多的主干。基本上,我只更改了模型,而没有更改架构,培训和测试yolov5。 因此,如果原始代码被更新,则更新该代码也非常方便。
目录
[培训与测试](#培训与测试)
特征
重组骨干,脖子,头部等模型结构,可以灵活方便地修改网络
更多主干,mobilenetV3,shufflenetV2,resnet18、50、101等
先决条件
请参阅requirements.txt
入门
数据集准备
以yolov5格式制作数据。 您可以使用utils / make_yolov5
2021-05-12 09:28:12
1.37MB
Python
1