# Official YOLOv7
Implementation of paper - [YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors](https://arxiv.org/abs/2207.02696)
[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/yolov7-trainable-bag-of-freebies-sets-new/real-time-object-detection-on-coco)](https://paperswithcode.com/sota/real-time-object-detection-on-coco?p=yolov7-trainable-bag-of-freebies-sets-new)
[![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/akhaliq/yolov7)
<a href="https://colab.research.google.com/gist/AlexeyAB/b769f5795e65fdab80086f6cb7940dae/yolov7detection.ipynb"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"></a>
[![arxiv.org](http://img.shields.io/badge/cs.CV-arXiv%3A2207.02696-B31B1B.svg)](https://arxiv.org/abs/2207.02696)
<div align="center">
<a href="./">
<img src="./figure/performance.png" width="79%"/>
</a>
</div>
## Web Demo
- Integrated into [Huggingface Spaces ����](https://huggingface.co/spaces/akhaliq/yolov7) using Gradio. Try out the Web Demo [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/akhaliq/yolov7)
## Performance
MS COCO
| Model | Test Size | AP<sup>test</sup> | AP<sub>50</sub><sup>test</sup> | AP<sub>75</sub><sup>test</sup> | batch 1 fps | batch 32 average time |
| :-- | :-: | :-: | :-: | :-: | :-: | :-: |
| [**YOLOv7**](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt) | 640 | **51.4%** | **69.7%** | **55.9%** | 161 *fps* | 2.8 *ms* |
| [**YOLOv7-X**](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7x.pt) | 640 | **53.1%** | **71.2%** | **57.8%** | 114 *fps* | 4.3 *ms* |
| | | | | | | |
| [**YOLOv7-W6**](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6.pt) | 1280 | **54.9%** | **72.6%** | **60.1%** | 84 *fps* | 7.6 *ms* |
| [**YOLOv7-E6**](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6.pt) | 1280 | **56.0%** | **73.5%** | **61.2%** | 56 *fps* | 12.3 *ms* |
| [**YOLOv7-D6**](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-d6.pt) | 1280 | **56.6%** | **74.0%** | **61.8%** | 44 *fps* | 15.0 *ms* |
| [**YOLOv7-E6E**](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6e.pt) | 1280 | **56.8%** | **74.4%** | **62.1%** | 36 *fps* | 18.7 *ms* |
## Installation
Docker environment (recommended)
<details><summary> <b>Expand</b> </summary>
``` shell
# create the docker container, you can change the share memory size if you have more.
nvidia-docker run --name yolov7 -it -v your_coco_path/:/coco/ -v your_code_path/:/yolov7 --shm-size=64g nvcr.io/nvidia/pytorch:21.08-py3
# apt install required packages
apt update
apt install -y zip htop screen libgl1-mesa-glx
# pip install required packages
pip install seaborn thop
# go to code folder
cd /yolov7
```
</details>
## Testing
[`yolov7.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7.pt) [`yolov7x.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7x.pt) [`yolov7-w6.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6.pt) [`yolov7-e6.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6.pt) [`yolov7-d6.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-d6.pt) [`yolov7-e6e.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6e.pt)
``` shell
python test.py --data data/coco.yaml --img 640 --batch 32 --conf 0.001 --iou 0.65 --device 0 --weights yolov7.pt --name yolov7_640_val
```
You will get the results:
```
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.51206
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.69730
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.55521
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.35247
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.55937
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.66693
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.38453
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.63765
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.68772
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.53766
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.73549
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.83868
```
To measure accuracy, download [COCO-annotations for Pycocotools](http://images.cocodataset.org/annotations/annotations_trainval2017.zip) to the `./coco/annotations/instances_val2017.json`
## Training
Data preparation
``` shell
bash scripts/get_coco.sh
```
* Download MS COCO dataset images ([train](http://images.cocodataset.org/zips/train2017.zip), [val](http://images.cocodataset.org/zips/val2017.zip), [test](http://images.cocodataset.org/zips/test2017.zip)) and [labels](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/coco2017labels-segments.zip). If you have previously used a different version of YOLO, we strongly recommend that you delete `train2017.cache` and `val2017.cache` files, and redownload [labels](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/coco2017labels-segments.zip)
Single GPU training
``` shell
# train p5 models
python train.py --workers 8 --device 0 --batch-size 32 --data data/coco.yaml --img 640 640 --cfg cfg/training/yolov7.yaml --weights '' --name yolov7 --hyp data/hyp.scratch.p5.yaml
# train p6 models
python train_aux.py --workers 8 --device 0 --batch-size 16 --data data/coco.yaml --img 1280 1280 --cfg cfg/training/yolov7-w6.yaml --weights '' --name yolov7-w6 --hyp data/hyp.scratch.p6.yaml
```
Multiple GPU training
``` shell
# train p5 models
python -m torch.distributed.launch --nproc_per_node 4 --master_port 9527 train.py --workers 8 --device 0,1,2,3 --sync-bn --batch-size 128 --data data/coco.yaml --img 640 640 --cfg cfg/training/yolov7.yaml --weights '' --name yolov7 --hyp data/hyp.scratch.p5.yaml
# train p6 models
python -m torch.distributed.launch --nproc_per_node 8 --master_port 9527 train_aux.py --workers 8 --device 0,1,2,3,4,5,6,7 --sync-bn --batch-size 128 --data data/coco.yaml --img 1280 1280 --cfg cfg/training/yolov7-w6.yaml --weights '' --name yolov7-w6 --hyp data/hyp.scratch.p6.yaml
```
## Transfer learning
[`yolov7_training.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7_training.pt) [`yolov7x_training.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7x_training.pt) [`yolov7-w6_training.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-w6_training.pt) [`yolov7-e6_training.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6_training.pt) [`yolov7-d6_training.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-d6_training.pt) [`yolov7-e6e_training.pt`](https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7-e6e_training.pt)
Single GPU finetuning for custom dataset
``` shell
# finetune p5 models
python train.py --workers 8 --device 0 --batch-size 32 --data data/custom.yaml --img 640 640 --cfg cfg/training/yolov7-custom.yaml --weights 'yolov7_training.pt' --name yolov7-custom --hyp data/hyp.scratch.custom.yaml
# finetune p6 models
python train_aux.py --workers 8 --device 0 --batch-size 16 --data data/custom.yaml --img 1280 1280 --cfg cfg/training/yolov7-w6-custom.yaml --weights 'yolov7-w6_training.pt' --name yolov7-w6-custom --hyp data/hyp.scratch.custom.yaml
```
## Re-parameterization
See [reparameterization.ipynb](tools/reparameterization.ipynb)
## Inference
On video:
``` shell
python detect.py --weights
没有合适的资源?快使用搜索试试~ 我知道了~
yolov7细分类型飞机-鸟类-无人机检测训练模型+数据集
共2000个文件
txt:1994个
pdf:3个
md:3个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
0 下载量 137 浏览量
2024-04-30
21:20:02
上传
评论
收藏 930.56MB ZIP 举报
温馨提示
yolov7细分类型飞机-鸟类-无人机检测训练模型, 包含1万多张yolo细分类型飞机-鸟类-无人机检测数据集,可区分具体飞机型号,数据集目录已经配置好,yolo格式(txt)的标签,划分好 train,val, test,并附有data.yaml文件,yolov5、yolov7、yolov8等算法可以直接进行训练模型, 数据集和检测结果参考: https://blog.csdn.net/zhiqingAI/article/details/136952543?spm=1001.2014.3001.5501
资源推荐
资源详情
资源评论
收起资源包目录
yolov7细分类型飞机-鸟类-无人机检测训练模型+数据集 (2000个子文件)
LICENSE.md 34KB
README.md 14KB
README.md 7KB
【yolov3-YOLOv5-yolov7-yolov8环境配置-教程1】.pdf 6.55MB
yolov7.pdf 5.85MB
【yolov3-YOLOv5-yolov7-yolov8环境配置-教程2】.pdf 580KB
Pexels-Videos-2068519_mp4-6_jpg.rf.152992a221524d15e4dc36a2ab9d647a.txt 3KB
Pexels-Videos-2068519_mp4-2_jpg.rf.7eef4182429fa5af9835b9e4b324326b.txt 3KB
Pexels-Videos-2068519_mp4-41_jpg.rf.fefe903e9abaf07e094996dea41e83a5.txt 3KB
production-ID_4714488_mp4-14_jpg.rf.522c1987b4c45360946de40283aac752.txt 3KB
Pexels-Videos-2068519_mp4-86_jpg.rf.3bfb181600d3d2ba1f264968fbef872d.txt 3KB
Pexels-Videos-2068519_mp4-53_jpg.rf.287fe40ed76e7fb02dd425239912407f.txt 3KB
Pexels-Videos-2068519_mp4-15_jpg.rf.334d03f1adad4f6aec8c6852173d5fdf.txt 3KB
Pexels-Videos-2068519_mp4-66_jpg.rf.0e4ca870655e5a5ba35ec359fdbb441f.txt 2KB
Pexels-Videos-2068519_mp4-92_jpg.rf.7ec61fada5202afcfedc716f71a621ed.txt 2KB
production-ID_5129465_mp4-5_jpg.rf.df732853d76b6ab89ff4c9c565f5ae9e.txt 1KB
Pexels-Videos-2068519_mp4-113_jpg.rf.4e1272c16f884769de2045382f9b0f6d.txt 1KB
a5905672d088c68ff92f0e5ea47e3e8c_jpg.rf.83f0969bdd213c831861799677514e6d.txt 1KB
production-ID_5129465_mp4-11_jpg.rf.2e3e531d63a0dcdf25828ec017870164.txt 1KB
pexels-magda-ehlers-5342234_mp4-72_jpg.rf.2b88c524f5a42065061562c68087e5d3.txt 1KB
pexels-magda-ehlers-5342234_mp4-78_jpg.rf.041973a78fc6d73ed140397c147b0729.txt 1012B
pexels-zuzanna-musial-5530954_mp4-3_jpg.rf.f2823bc615e80a86b7a02bc4be924fd8.txt 1002B
pexels-zuzanna-musial-5530954_mp4-1_jpg.rf.23c3e76396e85b1a4881fa656466bbba.txt 996B
d18d09f615587b9983a3f9b1e5137b1e_jpg.rf.4e2e8f15f11ff1e6bec2a81f3b317fa8.txt 992B
pexels-magda-ehlers-5342234_mp4-74_jpg.rf.8175e444a4fd18155e4a1294c1185236.txt 975B
pexels-magda-ehlers-5342234_mp4-77_jpg.rf.d2964da547a2ddec56cca695a6975600.txt 959B
bca331fb1eadbc15207b474c01c0a1f5_jpg.rf.e60570980c4cd2297ef6a57a09f3e55d.txt 935B
b863b4463a9048503b3d75d2d21e3467_jpg.rf.7b2a0f1a6f8508d5d63acd8021147877.txt 852B
pexels-yaroslav-shuraev-5419262_mp4-12_jpg.rf.057543d74c82b2163e32a493ff781f67.txt 816B
1e3645fab0dae766241d009199659fdf_jpg.rf.0d9d9bb0912d3ce0668c053a990e7ee0.txt 811B
video_mp4-11_jpg.rf.1af9e43bc62b58798b9cf29195720f60.txt 787B
pexels-cesar-a-ramirez-v-traphitho-5854976_mp4-1_jpg.rf.67d7d4752e835dd402d45a661db313c8.txt 766B
pexels-yaroslav-shuraev-5419262_mp4-16_jpg.rf.97942b8dfe4e0b7814a77a1407743404.txt 761B
Pexels-Videos-2068519_mp4-65_jpg.rf.0edfb3363d03405caf55c54b5b074865.txt 744B
pexels-magda-ehlers-5342234_mp4-42_jpg.rf.6317983024401e3414ffc3b9f88b03bc.txt 684B
57c6811d9712cd5e0068ca5ca7da76f6_jpg.rf.d848810e270dd8ec1ada2861ba3c0856.txt 629B
aaa92196435b3bf054c75edb594a7477_jpg.rf.10c18e96ad7bf66a9e7e97c330dce733.txt 626B
pexels-cesar-a-ramirez-v-traphitho-5854976_mp4-8_jpg.rf.d7ccc18a175e77e7aca91e50d35526bf.txt 621B
1fa31d2098941459b96dcd38a943a588_jpg.rf.9acbaf9456b1b0ef770f340612204203.txt 620B
pexels-magda-ehlers-5342234_mp4-58_jpg.rf.b2150eb212cf4e4bc9c136910cfcbdbf.txt 584B
pexels-manisha-shah-5560948_mp4-13_jpg.rf.b477edd8f319a5a7445da4247a0fb247.txt 584B
pexels-cesar-a-ramirez-v-traphitho-5854976_mp4-9_jpg.rf.ec61d3fb2b2af42671390575aa91af53.txt 576B
video--5-_mp4-9_jpg.rf.a5754b25bc68011d64aa263238271128.txt 552B
99b140181675966740d1c732c23b3a60_jpg.rf.07fc76e2bc9928128a3be32e2fadaf3f.txt 538B
1edff1136ba5a512927a43e72b146c50_jpg.rf.0c6bdcc3043eccd631b05b67c243cc67.txt 530B
production-ID_4982644_mp4-25_jpg.rf.defbd8bb3043b615f78630216bf236bc.txt 521B
pexels-cesar-a-ramirez-v-traphitho-5854976_mp4-10_jpg.rf.53b410f8fedb10cc44edebfae8b4e96d.txt 511B
131c595156acce09b0fafe191395f649_jpg.rf.6d8410628f9ea745602767c08e2470ba.txt 487B
fe5d4a267a08b36bff02d5571c276944_jpg.rf.c18698a77cdf25cd738b511dcace57d2.txt 487B
12cd0b326e3445d0c19f49c1cdfaaa5d_jpg.rf.7d047e62136d3f522d759fdb860232c3.txt 484B
pexels-nitesh-kamboj-5349616_mp4-2_jpg.rf.0cb3ac87b63bd796f11ed3f9a0813390.txt 482B
a6c083d4a5e9ff9be5c83f9a21441930_jpg.rf.bd9bd5701379aba97537cecd2f28e415.txt 471B
745c19e2f21d24d5522fc7cb3c04c76e_jpg.rf.0772e1b58e5d4599c75d759aefdc123e.txt 470B
add008b982e75f63b6af27022ea37cfb_jpg.rf.6659ef6cf654a55a3273e6170fe0a09d.txt 468B
6c3a84b42fe8429d0460859378e21e41_jpg.rf.a8dd2df81383eb4ae356f2d0be4b7415.txt 465B
pexels-yaroslav-shuraev-5419257_mp4-21_jpg.rf.f8ded2799ddb9d453016d03572bb9f17.txt 456B
00000125_jpg.rf.df33ed620461555f4c1caad16dfc7222.txt 451B
pexels-nitesh-kamboj-5349616_mp4-7_jpg.rf.78d1ee2648ffcd787c84b88e226e3659.txt 449B
Pexels-Videos-2068519_mp4-76_jpg.rf.91ee812a0fa902fe15b86b16a9c8ecd9.txt 443B
pexels-yaroslav-shuraev-5419262_mp4-3_jpg.rf.8bcce6e31d5134facc334323c9739875.txt 443B
0d34d8cfd25bfa7fe89fc08af0230641_jpg.rf.2b33448b925fd54608592a1ed67837a2.txt 443B
pexels-cesar-a-ramirez-v-traphitho-5854973_mp4-2_jpg.rf.9fb9035ca1fa578f6ce353e9e6c7abb4.txt 439B
9d5e5c2c86d525c2ce56b832a2deef4e_jpg.rf.e0213ce8344f774c321f4f43174c37be.txt 434B
pexels-nitesh-kamboj-5349616_mp4-3_jpg.rf.736e2b6565c49ffbc5cc86ad8a5465bf.txt 427B
pexels-cesar-a-ramirez-v-traphitho-5854971_mp4-7_jpg.rf.c6b8c4c891e43b6002ed7df6ee494373.txt 427B
d7c0fe9bb5416c7559984f1a5f436cd2_jpg.rf.ff3545bcdb7f2a716cb4a059dfdb2a3d.txt 410B
3a4bd7964915d7011d8833f64817aa88_jpg.rf.06ab227d306974487f4ac301e53f6fa0.txt 410B
pexels-cesar-a-ramirez-v-traphitho-5854973_mp4-9_jpg.rf.9daeed04243c378e58fcdff5dda86104.txt 407B
0fab6ced4bd8c68df368958e83ff6afe_jpg.rf.784df1632e5693779f54553268aca212.txt 406B
00001182_jpg.rf.4505fcb2139d52e04eeab803dfdc1f4e.txt 406B
3d23841d25007c85d13b2e663321ad39_jpg.rf.71cef027ec631027f033057fc5e03520.txt 405B
eebe263f7d4d5692c13ee2cab2850b28_jpg.rf.dc78c2722814bb5caf15b893297d6119.txt 403B
6629b589298caf995c01cd4e3f27c9c4_jpg.rf.707547be8a993abb1b70bf803d2bbeb1.txt 401B
pexels-cesar-a-ramirez-v-traphitho-5854971_mp4-9_jpg.rf.d44737e5ce36b8c37b7087304f2b352c.txt 397B
5ecfc397468fcf2ad8bc506b038c3760_jpg.rf.995b9947f379f5091cfff178e5a86dd9.txt 397B
a1a75b316b8f96412e02646eee7d3c22_jpg.rf.b07fecdd003f99240a23b005eb4ad794.txt 396B
45c9448c1b5a084e246111d55869e041_jpg.rf.d3cd09af920cee5d76d44d70a8e4e379.txt 394B
00000917_jpg.rf.325b4ef7a7451b46533527979471f5a3.txt 390B
04705e0911ce7e3dfebf94037d3acdb7_jpg.rf.1d9d3a5092eb6fabb5ea6b21085ac2c9.txt 390B
production-ID_4434372_mp4-2_jpg.rf.77e2bf7cc9866ee7c13ec2695403f3a9.txt 389B
pexels-egor-kunovsky-5922698_mp4-13_jpg.rf.0833f58040671c241d520b388c78ec1c.txt 387B
production-ID_4235128_mp4-6_jpg.rf.28370a51dfa4b325363c5596a77f8309.txt 385B
25975e7df056e744f020a7830deb6fe0_jpg.rf.9f615857feddee850b7189d8192af3a3.txt 384B
33dba0417321e0cb503c3d39b1dee41c_jpg.rf.595086df7513d87c97fd49f070646d83.txt 383B
250d93dc16622066f30cf7b642794ff2_jpg.rf.d0a4ad1899ab6968f8dbf8dcf6cee3ad.txt 378B
9c5b837ec4f7058b83ecf3a34bf8315e_jpg.rf.971e12f0ab7f8c64c119c2001b805da8.txt 373B
pexels-magda-ehlers-5342234_mp4-18_jpg.rf.1116d03f869eb00040f4023d597dcf65.txt 373B
32871edb08780a0863a90c285e123c87_jpg.rf.fd0cbb6c3762f6a3e4c5a58fa888878c.txt 372B
4b8fb38c35a84bfc2e0f83df66206986_jpg.rf.b06dc63c153c322f273afc2092bb7e21.txt 369B
production-ID_4235128_mp4-22_jpg.rf.0f67195c71a0d38506fdd47b79cb4a93.txt 364B
b0c8c5e726e00338348df4f0c72cc10f_jpg.rf.324e5ef473081146e4179043bbad9ec1.txt 328B
a8cb94c35f0d70f3c0d81e7c99c56d68_jpg.rf.620746667c3d3f6f6bed76e10787cf01.txt 327B
7f35bf982f5f6c207aaeee3abf134ace_jpg.rf.743a10d89690903ddb8d311360072f57.txt 326B
pexels-----6175266_mp4-1_jpg.rf.52b97e9504a7a4c1deeddbe210b09510.txt 326B
pexels-cesar-a-ramirez-v-traphitho-5854973_mp4-13_jpg.rf.8c374b9eff589f88d812ccfd8ed2b43b.txt 325B
3e8e9b33cae546f9cd705b04beb19b86_jpg.rf.63a85ede21447250ddb20b860ec07c64.txt 325B
b6b9bd5fc5de8594b7e7a5b261863db6_jpg.rf.ba5aaac5de7fb2647e8c30d73e3dc717.txt 324B
a0c8bc9659b9afa970465b96a8cfc9b3_jpg.rf.3a0774a1116966de4406f7e040f02337.txt 323B
5fbf1c97000553d77d58664b781c2729_jpg.rf.26aeeacdf5fa8938edb17d09c28576a3.txt 323B
b9e90c827310f174b8b9a17983f9680d_jpg.rf.a04e63fd3dba13228bcefac1f8a23561.txt 322B
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
stsdddd
- 粉丝: 2w+
- 资源: 688
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功