File size: 919 Bytes
f17d9c2 dad1dff f17d9c2 dad1dff 89032ba dad1dff 4fdccdd dad1dff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
---
license: gpl-3.0
inference: false
tags:
- object-detection
- computer-vision
- yolov6
- pypi
datasets:
- detection-datasets/coco
---
### Installation
```
pip install yolov6detect
```
### Yolov6 Inference
```python
from yolov6 import YOLOV6
model = YOLOV6(weights='kadirnar/yolov6s-v2.0', device='cuda:0')
model.classes = None
model.conf_thres = 0.25
model.iou_thresh = 0.45
model.view_img = False
model.save_img = 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}
}
``` |