# 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)
<img src="./figure/performance.png" height="480">
## 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>
```
# 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)
```
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
```
## Training
The training code and instrument will release soon.
## Citation
```
@article{wang2022yolov7,
title={{YOLOv7}: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors},
author={Wang, Chien-Yao and Bochkovskiy, Alexey and Liao, Hong-Yuan Mark},
journal={arXiv preprint arXiv:2207.02696},
year={2022}
}
```
## Acknowledgements
<details><summary> <b>Expand</b> </summary>
* [https://github.com/AlexeyAB/darknet](https://github.com/AlexeyAB/darknet)
* [https://github.com/WongKinYiu/yolor](https://github.com/WongKinYiu/yolor)
* [https://github.com/WongKinYiu/PyTorch_YOLOv4](https://github.com/WongKinYiu/PyTorch_YOLOv4)
* [https://github.com/WongKinYiu/ScaledYOLOv4](https://github.com/WongKinYiu/ScaledYOLOv4)
* [https://github.com/Megvii-BaseDetection/YOLOX](https://github.com/Megvii-BaseDetection/YOLOX)
* [https://github.com/ultralytics/yolov3](https://github.com/ultralytics/yolov3)
* [https://github.com/ultralytics/yolov5](https://github.com/ultralytics/yolov5)
</details>
没有合适的资源?快使用搜索试试~ 我知道了~
yolov7代码及6个预训练权重文件(训练文件全)
共61个文件
py:24个
yaml:23个
pt:5个
1.该资源内容由用户上传,如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
2.虚拟产品一经售出概不退款(资源遇到问题,请及时私信上传者)
版权申诉
5星 · 超过95%的资源 5 下载量 172 浏览量
2022-11-02
18:12:21
上传
评论 1
收藏 728.8MB ZIP 举报
温馨提示
该资源包含: 1.yolov7的代码 2.yolov7.pt yolov7x.pt yolov7-w6.pt yolov7-e6.pt yolov7-d6.pt yolov7-e6e.pt等6个预训练权重文件
资源推荐
资源详情
资源评论
收起资源包目录
yolov7代码及6个预训练权重文件.zip (61个子文件)
yolov7代码及6个预训练权重文件
yolov7-main
models
common.py 82KB
yolo.py 25KB
__init__.py 6B
export.py 4KB
experimental.py 4KB
cfg
baseline
yolor-csp-x.yaml 2KB
yolov3-spp.yaml 1KB
yolov3.yaml 1KB
r50-csp.yaml 1KB
yolor-csp.yaml 2KB
yolor-p6.yaml 2KB
yolor-d6.yaml 2KB
yolor-e6.yaml 2KB
yolov4-csp.yaml 2KB
x50-csp.yaml 1KB
yolor-w6.yaml 2KB
deploy
yolov7.yaml 4KB
yolov7-e6e.yaml 9KB
yolov7-w6.yaml 5KB
yolov7-e6.yaml 5KB
yolov7-tiny-silu.yaml 3KB
yolov7-d6.yaml 6KB
yolov7x.yaml 4KB
data
coco.yaml 1KB
hyp.scratch.p6.yaml 1KB
hyp.scratch.p5.yaml 1KB
hyp.scratch.tiny.yaml 1KB
test.py 17KB
train.py 36KB
detect.py 9KB
inference
images
horses.jpg 130KB
figure
performance.png 165KB
README.md 5KB
utils
loss.py 49KB
plots.py 18KB
metrics.py 9KB
general.py 32KB
datasets.py 55KB
activations.py 2KB
autoanchor.py 7KB
torch_utils.py 15KB
aws
userdata.sh 1KB
resume.py 1KB
mime.sh 780B
__init__.py 5B
__init__.py 6B
wandb_logging
log_dataset.py 815B
__init__.py 6B
wandb_utils.py 16KB
google_app_engine
additional_requirements.txt 105B
Dockerfile 821B
app.yaml 172B
google_utils.py 5KB
scripts
get_coco.sh 820B
hubconf.py 3KB
yolov7x.pt 136.47MB
yolov7-w6.pt 134.71MB
yolov7-main.zip 414KB
yolov7.pt 72.09MB
yolov7-d6.pt 255.83MB
yolov7-e6.pt 186.01MB
共 61 条
- 1
资源评论
- 未名v2024-03-27资源很受用,资源主总结的很全面,内容与描述一致,解决了我当下的问题。
- 阿粽2023-05-11资源有一定的参考价值,与资源描述一致,很实用,能够借鉴的部分挺多的,值得下载。
- Okay野象2023-03-06感谢资源主分享的资源解决了我当下的问题,非常有用的资源。
- m0_571816052023-04-21果断支持这个资源,资源解决了当前遇到的问题,给了新的灵感,感谢分享~
- 2301_809145022024-03-21资源内容详实,描述详尽,解决了我的问题,受益匪浅,学到了。
小风飞子
- 粉丝: 322
- 资源: 1500
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功