Skip to content
/ yoloair2 Public

☁️💡🎈专注于改进YOLOv7,Support to improve Backbone, Neck, Head, Loss, IoU, NMS and other modules

License

Notifications You must be signed in to change notification settings

iscyy/yoloair2

Repository files navigation

🌟 全新的YOLOv7改进版本来袭,最新改进点更新🚀2024

YOLOAir2☁️💡🎈 : Makes improvements easy again(专注于改进 YOLOv7 、YOLOv-Tiny 模型)

手把手改进YOLO系列: 全面改进篇更新🔗: 点击查看详情🚀 - 更多更全更新颖·《原创》·《最新》· 手把手改进YOLO系列详细教程(包括改进原理以及改进源代码 · 改进有效涨点)📚




YOLOAir2 算法库是一个基于 PyTorch 的 YOLO 系列算法组合工具箱。统一模型代码框架、统一应用、统一改进、易于模块组合、构建更强大的网络模型。

简体中文 | English

支持

https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair https://github.com/iscyy/yoloair

特性🚀使用🍉文档📒报告问题🌟更新💪讨论✌️效果预览🚀

https://github.com/iscyy/yoloair

Introduction

☁️💡🎈YOLOAir2 is the second version of the YOLOAir series, The framework is based on YOLOv7, including YOLOv7, YOLOv6, YOLOv5, YOLOX, YOLOR, YOLOv4, YOLOv3, Transformer, Attention and Improved-YOLOv7... Support to improve Backbone, Neck, Head, Loss, IoU, NMS and other modules, As a perfection and addition of YOLOAir

模型多样化: 基于不同网络模块构建不同检测网络模型。

模块组件化: 帮助用户自定义快速组合Backbone、Neck、Head,使得网络模型多样化,助力科研改进检测算法、模型改进,网络排列组合🏆。构建强大的网络模型。

统一模型代码框架、统一应用方式、统一调参、统一改进、集成多任务、易于模块组合、构建更强大的网络模型

内置集成YOLOv5、YOLOv7、YOLOv6、YOLOX、YOLOR、Transformer、PP-YOLO、PP-YOLOv2、PP-YOLOE、PP-YOLOEPlus、Scaled_YOLOv4、YOLOv3、YOLOv4、YOLO-Face、TPH-YOLO、YOLOv5Lite、SPD-YOLO、SlimNeck-YOLO、PicoDet等模型网络结构... 集成多种检测算法 和 相关多任务模型 使用统一模型代码框架,集成在 YOLOAir 库中,统一应用方式。便于科研者用于论文算法模型改进,模型对比,实现网络组合多样化。包含轻量化模型和精度更高的模型,根据场景合理选择,在精度和速度俩个方面取得平衡。同时该库支持解耦不同的结构和模块组件,让模块组件化,通过组合不同的模块组件,用户可以根据不同数据集或不同业务场景自行定制化构建不同检测模型。

支持集成多任务,包括目标检测、实例分割、图像分类、姿态估计、人脸检测、目标跟踪等任务

ingishvcn

Star🌟、Fork 不迷路,同步更新。

项目地址🌟: https://github.com/iscyy/yoloair

主要特性🚀

🚀支持更多的YOLO系列算法模型改进(持续更新...)

YOLOAir 算法库汇总了多种主流YOLO系列检测模型,一套代码集成多种模型:

  • 内置集成 YOLOv5 模型网络结构、YOLOv7 模型网络结构、 YOLOv6 模型网络结构、PP-YOLO 模型网络结构、PP-YOLOE 模型网络结构、PP-YOLOEPlus 模型网络结构、YOLOR 模型网络结构、YOLOX 模型网络结构、ScaledYOLOv4 模型网络结构、YOLOv4 模型网络结构、YOLOv3 模型网络结构、YOLO-FaceV2模型网络结构、TPH-YOLOv5模型网络结构、SPD-YOLO模型网络结构、SlimNeck-YOLO模型网络结构、YOLOv5-Lite模型网络结构、PicoDet模型网络结构等持续更新中...

Todo

内置网络模型配置支持✨

🚀包括基于 YOLOv5、YOLOv7、YOLOX、YOLOR、YOLOv3、YOLOv4、Scaled_YOLOv4、PPYOLO、PPYOLOE、PPYOLOEPlus、Transformer、YOLO-FaceV2、PicoDet、YOLOv5-Lite、TPH-YOLOv5、SPD-YOLO等其他多种改进网络结构等算法模型的模型配置文件


效果预览🚀

目标检测 目标分割
ingishvcn ingishvcn
图像分类 实例分割
ingishvcn ingishvcn
目标分割 目标跟踪
ingishvcn ingishvcn
姿态估计 人脸检测
ingishvcn ingishvcn
热力图01 热力图02
ingishvcn ingishvcn

yolo

预训练权重🚀


使用🍉

About the code. Follow the design principle of YOLOv7.
The original version was created based on YOLOv7 and YOLOAir

安装

Python>=3.7.0 的环境中克隆版本仓并安装 requirements.txt,包括PyTorch>=1.7

$ git clone https://github.com/iscyy/yoloair2.git  # 克隆
$ cd yoloair2
$ pip install -r requirements.txt  # 安装

训练

$ python train.py --cfg configs/yolov5/yolov5s.yaml

推理

detect.py 在各种数据源上运行推理, 并将检测结果保存到 runs/detect 目录。

$ python detect.py --source 0  # 网络摄像头
                          img.jpg  # 图像
                          vid.mp4  # 视频
                          path/  # 文件夹
                          path/*.jpg  # glob

Performance


YOLOv7训练教程✨

与YOLOv5框架基本一致,可以参考YOLOAir库


未来增强✨

后续会持续建设和完善 YOLOAir 生态
完善集成更多 YOLO 系列模型,持续结合不同模块,构建更多不同网络模型
横向拓展和引入关联技术等等


Citation✨

@article{2022yoloair2,
  title={{YOLOAir2}: Makes improvements easy again},
  author={iscyy},
  repo={github https://github.com/iscyy/yoloair2},
  year={2022}
}

Statement

Expand
  • The content of this site is only for sharing notes. If some content is infringing, please sending email.

  • If you have any question, please discuss with me by sending email.

Acknowledgements

Expand

https://github.com/ultralytics/yolov5
https://github.com/WongKinYiu/yolov7
https://github.com/iscyy/yoloair

About

☁️💡🎈专注于改进YOLOv7,Support to improve Backbone, Neck, Head, Loss, IoU, NMS and other modules

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages