Object Detection
computer-vision
yolov6
pypi
kadirnar commited on
Commit
a1d3005
1 Parent(s): 254d688

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -19,7 +19,7 @@ pip install yolov6detect
19
  ```python
20
  from yolov6 import YOLOV6
21
 
22
- model = YOLOV6(weights='yolov6s.pt', device='cuda:0')
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