Update README.md
Browse files
README.md
CHANGED
|
@@ -4,7 +4,7 @@ tags:
|
|
| 4 |
- object-detection
|
| 5 |
- computer-vision
|
| 6 |
- yolov6
|
| 7 |
-
-
|
| 8 |
datasets:
|
| 9 |
- detection-datasets/coco
|
| 10 |
---
|
|
@@ -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
|
|
|
|
| 4 |
- object-detection
|
| 5 |
- computer-vision
|
| 6 |
- yolov6
|
| 7 |
+
- yolo
|
| 8 |
datasets:
|
| 9 |
- detection-datasets/coco
|
| 10 |
---
|
|
|
|
| 25 |
```python
|
| 26 |
from yolov6 import YOLOV6
|
| 27 |
|
| 28 |
+
model = YOLOV6(weights='kadirnar/yolov6m-v3.0', device='cuda:0', hf_model=True)
|
| 29 |
|
| 30 |
model.classes = None
|
| 31 |
model.conf = 0.25
|