# 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
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
yolo7中大小各种尺度红外无人机检测,训练好的权重,可以直接使用,并附有8000左右yolo小型型固定翼无人机检测数据集,数据集目录已经配置好,yolo格式(txt)的标签,划分好 train,val, test,并附有data.yaml文件,yolov5、yolov7、yolov8等算法可以直接进行训练模型, 数据集和检测结果参考: https://blog.csdn.net/zhiqingAI/article/details/124230743 https://blog.csdn.net/zhiqingAI/article/details/136952543 数据集配置目录结构data.yaml: nc: 1 names: ['IR_Drone']
资源推荐
资源详情
资源评论
收起资源包目录
YOLOv7算法各种尺度红外无人机检测权重+8000红外无人机数据集+使用教程 (2000个子文件)
LICENSE.md 34KB
README.md 14KB
README.md 7KB
【yolov3-YOLOv5-yolov7-yolov8环境配置-教程2】.md 5KB
【yolov3-YOLOv5-yolov7-yolov8环境配置-教程1】.pdf 6.55MB
yolov7.pdf 5.85MB
【yolov3-YOLOv5-yolov7-yolov8环境配置-教程2】.pdf 580KB
loss.py 73KB
datasets.py 55KB
general.py 36KB
plots.py 20KB
wandb_utils.py 16KB
torch_utils.py 15KB
metrics.py 9KB
export.py 9KB
autoanchor.py 7KB
add_nms.py 5KB
google_utils.py 5KB
hubconf.py 3KB
activations.py 2KB
resume.py 1KB
log_dataset.py 815B
__init__.py 6B
__init__.py 6B
__init__.py 5B
userdata.sh 1KB
mime.sh 780B
results.txt 15KB
pic_323_jpgrfa9ae7f57dbe50eb66c5b390f21225c76_jpg.rf.ba68bd82f40ca8905a815dd3bc39187d.txt 395B
234_jpgrfa37cdd2a7050c2786f17385f55cb7a62_jpg.rf.6d978e9bd7f2aeb47ea3f6baa1a09127.txt 362B
5_pngrf4b4e61b4669b2015b92dcf596ef60fbe_jpg.rf.8ee362e167ca4791b9be9225f58417c8.txt 329B
0001_jpgrfaf820ca2a92fb2c4372b6cdc9254d056_jpg.rf.8a2db82c8e21c8faba2af8d4a920912c.txt 273B
8_jpgrf11218a2a74b3e34f85c8f7bd32be5ba8_jpg.rf.09e5e08470a23e8bbce54ede476dbaf7.txt 267B
782_pngrf20abc61101c9d8ef64adf79196c06c77_jpg.rf.c867402ec7b3318a92d7c20999d1a61e.txt 184B
782_pngrf20abc61101c9d8ef64adf79196c06c77_jpg.rf.1c4dce61667785041101af21903701ca.txt 162B
使用说明.txt 152B
pic_935_jpgrf39b28c98a2178eab406398f0c7fb8837_jpg.rf.7fb6e7cd6cc4ab15aebff241d4cc231b.txt 135B
16_pngrf09571ce57bdcf2e63a4899a6eac241ca_jpg.rf.9613a05a6ee3025fb2f8fa27a6648cf9.txt 132B
pic_218_jpgrf039d39aa161eec99e6e61de019b12e3e_jpg.rf.c69255452b50bea8c7755739f9d716c9.txt 116B
0090_jpgrf19b0a6272ee70076b6ceac224f2625f1_jpg.rf.f3612be3ac07045088cb91afcabbbc65.txt 109B
additional_requirements.txt 105B
pic_789_jpgrfd8292e17779976da59007127ad0e127b_jpg.rf.92a2a7362928811faafa8cdc51ff7839.txt 100B
0312_jpgrff757ad14a0d461b26a80e7f674c1ca7d_jpg.rf.e0458577274fbb096b48747aca44430e.txt 96B
0090_jpgrf19b0a6272ee70076b6ceac224f2625f1_jpg.rf.51ab14ea96f8a0998890a0388612c0b2.txt 92B
pic_116_jpgrf3f3542757f7e43ef1cf454afe3da5b9e_jpg.rf.255e896d69f029bd64187b30ba0904fd.txt 81B
784_pngrfa814b14f86e32cc7b1242e7a446f7243_jpg.rf.2f3748786b53af87dd7bb136a77f8f45.txt 81B
192_jpgrf35519a51c719d8d80fd0e5fefcd8d2a1_jpg.rf.792ec2e6dfa0382ba78d9f029e4334ee.txt 81B
41_pngrf2cc2a916171b6f854f10a0641d98bad6_jpg.rf.965777d0f332a8df6d40e2d1c5429298.txt 81B
339_jpgrf791202cfe9e2994331da2ea40c6a793c_jpg.rf.6fdb6a18c6fa9b80d9df94f6d7af0317.txt 80B
302_jpgrfd392c02dfcb97f0d258a7d7fb30e3c7a_jpg.rf.e869ea9c50818268d4bd87a604aa7e81.txt 80B
278_jpgrfa79107bce086c2ddc67ddd1498a407f6_jpg.rf.90a69fe28b6e94e0d6aea0278fc7ed54.txt 80B
281_jpgrfd17cb356d11f0b5409d40a37b632400b_jpg.rf.aced063de4037972656e60601261bc80.txt 80B
247_jpgrf6b6b804dbf5c1090aa976b0c8c178047_jpg.rf.4f18a9d82e9af73c44421aedbb88c81a.txt 80B
304_jpgrf639f29738ac7a3fb0a08bf8f0d5e7191_jpg.rf.d92f7f4b7b56946d730deeee5939fe19.txt 79B
72_jpgrf002fcffa86749a1f4139e44c6d065cc7_jpg.rf.4e8b65d629ff8ca06885123446ddc3ed.txt 79B
72_jpgrf002fcffa86749a1f4139e44c6d065cc7_jpg.rf.58e3d7beebad3ed0aa5e05c1354a0626.txt 79B
247_jpgrf6b6b804dbf5c1090aa976b0c8c178047_jpg.rf.b268a7470d803b1cbc6964b3eff0ec47.txt 78B
0225_jpgrf8f0b4b92d9bdf7b4bec9a3895f8616cb_jpg.rf.eb3c9ef5c282f4c01defc33bd47182c2.txt 78B
0324_jpgrff82a947a0618c16ae1f5647305065e06_jpg.rf.c958df5000c5534f9bf38edc6bd802e7.txt 78B
24_jpgrf764e5b59901f16d6812d0091ea81b2b4_jpg.rf.dfb2f3904eb131ec4d7548b52abffd38.txt 78B
427_jpgrf3187ebbb6c5ffb8338ea08640f82d4fb_jpg.rf.47e0c0fbbcb6ed8d075cd1b8e33ea3f4.txt 78B
0239_jpgrfa176730502a63577488cd9820a494710_jpg.rf.3c1d5c399b00f1de834de7c9d3596ff8.txt 78B
0287_jpgrfbfdc382213b2778ec1bf2d2060d6595f_jpg.rf.1ad2602f876314831061876f263c7ac7.txt 78B
pic_609_jpgrfb31d1e7fa5253d6b44aba564bbcd0a8f_jpg.rf.d5a86a04645e056ddbebc1c045b08a5f.txt 78B
pic_730_jpgrf7d6e43606e3815cbb37680f02f6895e9_jpg.rf.b5121e24563504714832419e1d4be9b4.txt 78B
24_jpgrf764e5b59901f16d6812d0091ea81b2b4_jpg.rf.0bc3a46f5d97a243d877a2c4366798dc.txt 78B
252_jpgrf494a40354228dcf2a81b831ab7df8fdd_jpg.rf.6bc520bcca86278511ce8bd4b3814cf1.txt 78B
pic_569_jpgrf727b870bfd3e3cf653b2efc419e8a0ba_jpg.rf.4a15a2df66101eb74adff1ba57c11cac.txt 78B
64_jpgrf7383eb72801c40d53a5738826b92583f_jpg.rf.638522ddbd3ee8efc4bd23d28bd42e6f.txt 78B
43_jpgrf6f4caa902ae91edccf55d3f7e18612dd_jpg.rf.ddcb8555f1e2189cedfda98247ff3906.txt 78B
pic_569_jpgrf727b870bfd3e3cf653b2efc419e8a0ba_jpg.rf.46c4f6ff4ce4df169661c1c9f08c4055.txt 78B
pic_169_jpgrfa050f09fbeb6600df77b7fad7ce40ec5_jpg.rf.c63af8c6780afb767809b0785a2d6e1a.txt 78B
pic_229_jpgrf4b155a4232316bca934557e95410889c_jpg.rf.8dee9c9a16a12e308b6c6f9074079002.txt 78B
32_jpgrfafec614fa504566387fd190ac1d4d2f4_jpg.rf.39ebeded361829c26f9b4123d559cafc.txt 77B
716_pngrfa2a90c21b5d2bcf81cff5f5714896714_jpg.rf.9e48c5b05375235eb18eba7c184093db.txt 77B
pic_515_jpgrf9b1de08306107f8447192090a810a4e3_jpg.rf.c8b6608b23f1e367113556d62430dc3f.txt 77B
447_jpgrf862e5d98cfbe7503f429ef2553618886_jpg.rf.b606847b84a261fb012c7e0660c03ee8.txt 77B
pic_285_jpgrf8cc2cf1d9d44c36a064fd9edc44f3a9c_jpg.rf.ee462ca3821ad6ddb388e4b9d9de088e.txt 77B
716_pngrfa2a90c21b5d2bcf81cff5f5714896714_jpg.rf.784f00a433110babd7ed2b3a696b89d3.txt 77B
51_jpgrfbae52778a149277ceae197785d77b243_jpg.rf.c8f7ada11b571cd6fe71c61ffa8d90ec.txt 77B
795_pngrf8a01c5014bff97644aefe7e5dd0f9b4c_jpg.rf.4805bd44c006a6f1976cf0a173adf9b1.txt 77B
pic_770_jpgrf74a533591b36229e1365da7ba8884b41_jpg.rf.12d219f97a9977750fd031a8c6031d28.txt 77B
302_jpgrfd392c02dfcb97f0d258a7d7fb30e3c7a_jpg.rf.39bc473e4445c79a43247e94132c1d73.txt 77B
43_jpgrf6f4caa902ae91edccf55d3f7e18612dd_jpg.rf.0cafa025927e3020d274cb9668dfc59b.txt 77B
369_jpgrf5670c836196b983d922e0b7fa70e3a4e_jpg.rf.4b8d9019ff468e8cabc06b35db30602d.txt 77B
0030_jpgrf47f91190d2755f78884328b7bc15c88d_jpg.rf.16361138ac94655eee0d5df6af584ff4.txt 77B
262_jpgrf99e30a457236488894b05d7364fbb4c3_jpg.rf.2a6eaad2fb1fcd05929cea833ffaa90e.txt 77B
806_pngrfec4c2171cc72dc99e93b845b3772500d_jpg.rf.63f5c69716f57250c5f483d9cad8f113.txt 77B
0072_jpgrf7c2116d751a249ae05e0efca453133d7_jpg.rf.e39a4a6e2901021868ade5231650c929.txt 77B
261_jpgrf1a774b86a50733169418cb19494908a6_jpg.rf.04ac75344f177acbe71d21323b0ef57e.txt 77B
795_pngrf8a01c5014bff97644aefe7e5dd0f9b4c_jpg.rf.7176f482bb17d6a97de03a18a78c6cf5.txt 76B
261_jpgrf1a774b86a50733169418cb19494908a6_jpg.rf.f85cd176b6ee36c93c62846791c2147c.txt 76B
0042_jpgrff33221a99a3711829442da0ef85875f4_jpg.rf.1d60354769c1158043d1a9dc6516b9b4.txt 76B
pic_367_jpgrfc4b4602499ca7f0214f1e0ceeb601700_jpg.rf.d4ad08b05872fc9f54f224e89bd6873d.txt 76B
pic_337_jpgrfad9c75f1ab5878b942ebb45ea23c5ba2_jpg.rf.195f1e40ca176d53ec29e002ba283063.txt 76B
32_jpgrfafec614fa504566387fd190ac1d4d2f4_jpg.rf.75b63db11d113176a1cacb91bd70a00f.txt 76B
pic_262_jpgrf20b530bb7ef6c7043e7c1fffdaff1e8e_jpg.rf.3b9918e9c40dafe66788ebe97fcca85a.txt 76B
169_jpgrf4cd5c28a87eed4121bad7a606215f512_jpg.rf.03af3cb2a5b32af26d18825f38fb3dd5.txt 73B
41_pngrf2cc2a916171b6f854f10a0641d98bad6_jpg.rf.0b4e7997ca0e8de9e75ba8d29ce49b30.txt 72B
78_jpgrf5c8ee5f278561af9bc6303f55b0dfde2_jpg.rf.9d294d8d6685f6948a68b6688a5137c9.txt 72B
共 2000 条
- 1
- 2
- 3
- 4
- 5
- 6
- 20
资源评论
stsdddd
- 粉丝: 2w+
- 资源: 710
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功