Top AI Repos — open-source AI, indexed and scored
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam，dcn and so on), and tensorrt
| Date | Stars |
|---|---|
| 2026-07-24 | 692 |
| 2026-07-25 | 692 |
| 2026-07-28 | 692 |
| 2026-07-30 | 692 |
| 2026-08-06 | 692 |
Today
— stars today
This week
— stars this week
This month
— stars this month
Momentum
0.0
growth rate 0.00%/day
# flexible-yolov5
<div align=center><img src="https://user-images.githubusercontent.com/9928596/191880047-a408e29c-4c6b-4e20-9318-0b23037ed64e.png" width="600" height="400"/></div>
*Update the code for [ultralytics/yolov5](https://github.com/ultralytics/yolov5) version 6.1.*
---
代码基于U版YOLOv5 6.1版本. 根据 {backbone, neck, head} 重新组织了网络结构, 目前backbone 除了原始的YOLO外，还可选择 resnet, hrnet, swin-transformer, gnn, mobilenet 等主流backbone. 同时也可以自由的加入 SE, DCN, drop block 等插件. 可以很方便的对网络结构等进行替换、修改、实验. 同时提供了tensorrt 的c++、Python 推理, 量化. 以及Triton、tf_serving 部署代码. 每个backbone只选了一个训练300个epoch做对比，均无预训练权重，由于网络结构不同，我的结果并不能代表网络最终的结果，可以作为一个baseline参考. 这个项目适合想要各种改YOLO或者验证模块. 是如果你有什么好的idea，比如增加新的backbone, 插件等, 欢迎提PR, 使用时遇到什么问题, 也欢迎提issue. 如果对你有帮助, 感谢给颗♥(ˆ◡ˆԅ)小 ⭐️⭐️.
---
Split the yolov5 model to {backbone, neck, head} to facilitate the operation of various modules and support more backbones.Basically, only change the model, and I didn't change the architecture, training and testing of yolov5. Therefore, if the original code is updated, it is also very convenient to update this code. if you have some new ideas, you can give a pull request, add new features together。 if this repo can help you, please give me a star.
## Table of contents
* [Features](#features)
* [Notices](#Notices)
* [Bugs](#Bugs)
* [Prerequisites](#prerequisites)
* [Getting Started](#getting-started)
* [Dataset Preparation](#dataset-preparation)
* [Training and Testing](#Training-and-Testing)
* [Training](#training)
* [Testing and Visualize](#testing-and-visualize)
* [Model performance comparison](#Model-performance-comparison-with-different-backbone)
* [Detection](#Detection)
* [Deploy](#Deploy)
* [Export](#Export)
* [Grpc server](#Grpc-server)
* [Reference](#Reference)
## Features
- support nvidia tensor-core 4:2 sparsity.
- support QAT, for qat onnx export, need torch >=1.13(I only test on this version) 2023-10-17
- update PTQ code , 2023-10-01
- Reorganize model structure, such as backbone, neck, head, can modify the network flexibly and conveniently
- mobilenetV3-small, mobilenetV3-large
- shufflenet_v2_x0_5, shufflenet_v2_x1_0, shufflenet_v2_x1_5, shufflenet_v2_x2_0
- yolov5s, yolov5m, yolov5l, yolov5x, yolov5transformer
- resnet18, resnet50, resnet34, resnet101, resnet152
- efficientnet_b0 - efficientnet_b8, efficientnet_l2
- hrnet 18,32,48
- CBAM, SE
- swin transformer - base, tiny, small, large (please set half=False in scripts/eval.py and don't use model.half in train.py)
- DCN (mixed precision training not support, if you want use dcn, please close amp in line 292 of scripts/train.py)
- coord conv
- drop_block
- vgg, repvgg
- tensorrt c++/python infer, triton server infer
- gnn backbone
## Notices
* The CBAM, SE, DCN, coord conv. At present, the above plug-ins are not added to all networks, so you may need to modify the code yourself.
* The default gw and gd for PAN and FPN of other backbone are same as yolov5_s, so if you want a strong model, please modify self.gw and self.gd in FPN and PAN.
* resnet with dcn, training on gpu *RuntimeError: expected scalar type Half but found Float: please remove the mixed precision training in line 351 of scripts/train.py
* swin-transformer, training is ok, but testing report *RuntimeError: expected object of scalar type Float but got scalar type Half for argument #2 'mat2' in call to_th_bmm_out in swin_trsansformer.py. please set half=False in script/eval.py
* mobilenet export onnx failed, please replace HardSigmoid() by others, because onnx don't support pytorch nn.threshold
## Bugs
None
## Prerequisites
please refer requirements.txt
## Getting Started
### Dataset Preparation
Make data for yolov5 format. you can use od/data/transform_voc.py convert VOC data to yolov5 data format.
### Training and Testing
For training and Testing, it's same like yolov5.
#### Training
1. check out configs/data.yaml, and replace with your data， and number of object nc
2. check out configsExcerpt of 8,871 characters
Read on GitHub
123
2
1
Would you bet a product on this? Bounded 0–100 and slow moving.
matched fp:cc6e7da45553f5b9, topic:sparsity, readme:sparsity
matched fp:cc6e7da45553f5b9, topic:tensorrt
matched fp:cc6e7da45553f5b9, topic:pytorch
matched fp:cc6e7da45553f5b9, topic:object-detection