kadirnar commited on
Commit
baa93a6
1 Parent(s): da18eef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +49 -0
README.md CHANGED
@@ -1,3 +1,52 @@
1
  ---
2
  license: gpl-3.0
 
 
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: gpl-3.0
3
+ tags:
4
+ - object-detection
5
+ - computer-vision
6
+ - yolor
7
+ - yolov4
8
+ datasets:
9
+ - detection-datasets/coco
10
+ language:
11
+ - en
12
+ pipeline_tag: object-detection
13
+ library_name: yolor
14
+ library_version: 0.2.3
15
  ---
16
+
17
+ ### Model Description
18
+ [YOLOR:](https://arxiv.org/abs/2105.04206) You Only Learn One Representation: Unified Network for Multiple Tasks.
19
+
20
+ [YOLOR-Pip:](https://github.com/kadirnar/yolor-pip/) Packaged version of the YOLOR repository
21
+
22
+ [Paper Repo:](https://github.com/WongKinYiu/yolor/) Implementation of paper - YOLOR
23
+
24
+ ### Installation
25
+ ```
26
+ pip install yolor
27
+ ```
28
+
29
+ ### Yolov6 Inference
30
+ ```python
31
+ from yolor.helpers import Yolor
32
+
33
+ model = Yolor(cfg='yolor/cfg/yolor_p6.cfg', weights='kadirnar/yolor-p6', imgsz=640, device='cuda:0')
34
+
35
+ model.classes = None
36
+ model.conf = 0.25
37
+ model.iou_ = 0.45
38
+ model.show = False
39
+ model.save = True
40
+
41
+ model.predict('yolor/data/highway.jpg')
42
+ ```
43
+
44
+ ### BibTeX Entry and Citation Info
45
+ ```
46
+ @article{wang2021you,
47
+ title={You Only Learn One Representation: Unified Network for Multiple Tasks},
48
+ author={Wang, Chien-Yao and Yeh, I-Hau and Liao, Hong-Yuan Mark},
49
+ journal={arXiv preprint arXiv:2105.04206},
50
+ year={2021}
51
+ }
52
+ ```