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
|