YOLOAir 算法代码库是一个基于 PyTorch 的 YOLO 系列目标检测开源工具箱。使用统一模型代码框架、统一应用方式、统一调参,该库包含大量的改进模块,可使用不同网络模块来快速构建不同网络的检测模型。基于 YOLOv5 代码框架,并同步适配 YOLOv5(v6.0/v6.1 更新) 部署生态。用户在使用这个项目之前, 可以先了解 YOLOv5 库。
该项目包含大量的改进方式,并能降低改进难度,改进点包含 Backbone、Neck、Head、注意力机制、IoU 损失函数、多种 NMS、Loss 损失函数、自注意力机制系列、数据增强部分、激活函数等部分,更多内容可以关注 YOLOAir 项目的说明文档。项目同时附带各种改进点原理及对应的代码改进方式教程,用户可根据自身情况快速排列组合,在不同的数据集上实验, 应用组合改进点写论文!
模块组件化:帮助用户自定义快速组合 Backbone、Neck、Head,使得网络模型多样化,使得改进检测算法、工程算法部署落地更便捷,构建更强大的网络模型。
支持YOLOv5、YOLOv7、YOLOX、YOLOR、YOLOv3、YOLOv4、Scaled_YOLOv4、Transformer等算法网络模型进行改进。
项目地址: https://github.com/iscyy/yoloair
项目介绍
主要特性
持续更新支持更多的 YOLO 系列算法模型,作者对可以进行改进的部分进行了分类:
支持更多 Backbone
支持更多 Neck
支持更多检测头 Head
支持更多即插即用的注意力机制
支持更多损失函数
支持更多 NMS
支持更多数据增强
支持更多 Loss
内置多种网络模型模块化组件
以上组件模块使用统一模型代码框架、统一任务形式、统一应用方式,模块组件化可以帮助用户自定义快速组合 Backbone、Neck、Head,使得网络模型多样化,助力科研改进检测算法,构建更强大的网络模型。
内置改进网络模型配置支持
包括基于 YOLOv5 的其他几十种改进网络结构等算法模型的 yaml 配置文件汇总,使用 YOLOv5 网络作为示范,可以将这些模块无缝加入到 YOLOv7、YOLOX、YOLOR、YOLOv4、Scaled_YOLOv4、YOLOv3 等系列 YOLO 算法模块。
用户可自行基于提供的网络模块,进行自定义改进网络。
YOLOAir 算法库汇总了多种主流 YOLO 系列检测模型,一套代码汇集多种模型结构:
以上多种检测算法网络模型使用统一代码框架,集成在 YOLOAir 代码库中,统一应用方式。便于科研者用于论文算法模型改进,模型对比,实现网络组合多样化,包含轻量化模型和精度更高的模型,根据场景合理选择,在精度和速度者两个方面取得平衡。同时该库支持解耦不同的结构和模块组件,让模块组件化,通过组合不同的模块组件,用户可以根据不同数据集或不同业务场景自行定制化构建不同检测模型。
使用
代码遵循 YOLOv5 设计原则,使用方式基本和 YOLOv5 框架对齐。
安装
在 python>=3.7.0 的环境中克隆版本仓并安装 requirements.txt,包括 PyTorch>=1.7。
$ git clone https://github.com/iscyy/yoloair.git # 克隆
$ cd yoloair
$ pip install -r requirements.txt # 安装
训练
$ python train.py --data coco128.yaml --cfg configs/yolov5/yolov5s.yaml
推理
detect.py 在各种数据源上运行推理, 并将检测结果保存到 runs/detect 目录。
$ python detect.py --source img.jpg
融合
如果使用不同的模型推理数据集,可以使用 wbf.py 通过加权框融合来集成结果。在 wbf.py 中设置 img 路径和 txt 路径。
$ python tools/wbf.py
教程
基本教程与 YOLOv5 框架同步,
其他教程如下 (持续更新中…)