Update README.md
Browse files
README.md
CHANGED
@@ -19,7 +19,7 @@ pip install yolov6detect
|
|
19 |
```python
|
20 |
from yolov6 import YOLOV6
|
21 |
|
22 |
-
model = YOLOV6(weights='
|
23 |
|
24 |
model.classes = None
|
25 |
model.conf_thres = 0.25
|
|
|
19 |
```python
|
20 |
from yolov6 import YOLOV6
|
21 |
|
22 |
+
model = YOLOV6(weights='kadirnar/yolov6n-v2.0', device='cuda:0')
|
23 |
|
24 |
model.classes = None
|
25 |
model.conf_thres = 0.25
|