README.md

---
license: gpl-3.0
tags:
  - object-detection
  - computer-vision
  - yolov6
  - pypi
datasets:
  - detection-datasets/coco
---

### Model Description
[YOLOv6:](https://arxiv.org/abs/2209.02976) A single-stage object detection framework dedicated to industrial applications.

[YOLOv6 v3.0](https://arxiv.org/abs/2301.05586): A Full-Scale Reloading

[YOLOv6-Pip: Packaged version of the Yolov6 repository](https://github.com/kadirnar/yolov6-pip/)

[Paper Repo: Implementation of paper - YOLOv6](https://github.com/meituan/YOLOv6/)

### Installation
```
pip install yolov6detect
```

### Yolov6 Inference
```python
from yolov6 import YOLOV6

model = YOLOV6(weights='kadirnar/yolov6n-v2.0', device='cuda:0',hf_model=True)

model.classes = None
model.conf = 0.25
model.iou = 0.45
model.show = False
model.save = True

pred = model.predict(source='data/images',yaml='data/coco.yaml', img_size=640)
```

### BibTeX Entry and Citation Info
 ```
@article{li2022yolov6,
  title={YOLOv6: A single-stage object detection framework for industrial applications},
  author={Li, Chuyi and Li, Lulu and Jiang, Hongliang and Weng, Kaiheng and Geng, Yifei and Li, Liang and Ke, Zaidan and Li, Qingyuan and Cheng, Meng and Nie, Weiqiang and others},
  journal={arXiv preprint arXiv:2209.02976},
  year={2022}
}
```