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_p6.cfg',
|
| 35 |
+
weights='kadirnar/yolor-p6',
|
| 36 |
+
imgsz=640,
|
| 37 |
+
device='cuda:0'
|
| 38 |
+
)
|
| 39 |
|
| 40 |
model.classes = None
|
| 41 |
model.conf = 0.25
|