Update README.md
Browse files
README.md
CHANGED
@@ -11,7 +11,7 @@ language:
|
|
11 |
- en
|
12 |
pipeline_tag: object-detection
|
13 |
library_name: yolor
|
14 |
-
library_version: 0.
|
15 |
---
|
16 |
|
17 |
### Model Description
|
@@ -30,7 +30,12 @@ pip install yolor
|
|
30 |
```python
|
31 |
from yolor.helpers import Yolor
|
32 |
|
33 |
-
model = Yolor(
|
|
|
|
|
|
|
|
|
|
|
34 |
|
35 |
model.classes = None
|
36 |
model.conf = 0.25
|
|
|
11 |
- en
|
12 |
pipeline_tag: object-detection
|
13 |
library_name: yolor
|
14 |
+
library_version: 0.0.3
|
15 |
---
|
16 |
|
17 |
### Model Description
|
|
|
30 |
```python
|
31 |
from yolor.helpers import Yolor
|
32 |
|
33 |
+
model = Yolor(
|
34 |
+
cfg='yolor/cfg/yolor_w6.cfg',
|
35 |
+
weights='kadirnar/yolor-w6',
|
36 |
+
imgsz=640,
|
37 |
+
device='cuda:0'
|
38 |
+
)
|
39 |
|
40 |
model.classes = None
|
41 |
model.conf = 0.25
|