Object Detection
computer-vision
yolov6
yolo
kadirnar commited on
Commit
c52ff09
1 Parent(s): 1307c69

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +44 -0
README.md CHANGED
@@ -1,3 +1,47 @@
1
  ---
2
  license: gpl-3.0
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: gpl-3.0
3
+ tags:
4
+ - object-detection
5
+ - computer-vision
6
+ - yolov6
7
+ - yolo
8
+ datasets:
9
+ - detection-datasets/coco
10
  ---
11
+
12
+ ### Model Description
13
+ [YOLOv6: A single-stage object detection framework dedicated to industrial applications.](https://arxiv.org/abs/2209.02976)
14
+
15
+ [YOLOv6-Pip: Packaged version of the Yolov6 repository](https://github.com/kadirnar/yolov6-pip/)
16
+
17
+ [Paper Repo: Implementation of paper - YOLOv6](https://github.com/meituan/YOLOv6/)
18
+
19
+ ### Installation
20
+ ```
21
+ pip install yolov6detect
22
+ ```
23
+
24
+ ### Yolov6 Inference
25
+ ```python
26
+ from yolov6 import YOLOV6
27
+
28
+ model = YOLOV6(weights='kadirnar/yolov6l-v3.0', device='cuda:0', hf_model=True)
29
+
30
+ model.classes = None
31
+ model.conf = 0.25
32
+ model.iou = 0.45
33
+ model.show = False
34
+ model.save = True
35
+
36
+ pred = model.predict(source='data/images',yaml='data/coco.yaml', img_size=640)
37
+ ```
38
+
39
+ ### BibTeX Entry and Citation Info
40
+ ```
41
+ @article{li2022yolov6,
42
+ title={YOLOv6: A single-stage object detection framework for industrial applications},
43
+ 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},
44
+ journal={arXiv preprint arXiv:2209.02976},
45
+ year={2022}
46
+ }
47
+ ```