Update README.md
Browse files
README.md
CHANGED
@@ -25,7 +25,7 @@ pip install yolov6detect
|
|
25 |
```python
|
26 |
from yolov6 import YOLOV6
|
27 |
|
28 |
-
model = YOLOV6(weights='kadirnar/
|
29 |
|
30 |
model.classes = None
|
31 |
model.conf = 0.25
|
|
|
25 |
```python
|
26 |
from yolov6 import YOLOV6
|
27 |
|
28 |
+
model = YOLOV6(weights='kadirnar/yolov6s6-v3.0', device='cuda:0', hf_model=True)
|
29 |
|
30 |
model.classes = None
|
31 |
model.conf = 0.25
|