Object Detection
computer-vision
yolov6
pypi
kadirnar commited on
Commit
4195310
1 Parent(s): 9267d4a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -25,13 +25,13 @@ pip install yolov6detect
25
  ```python
26
  from yolov6 import YOLOV6
27
 
28
- model = YOLOV6(weights='kadirnar/yolov6s-v2.0', device='cuda:0')
29
 
30
  model.classes = None
31
- model.conf_thres = 0.25
32
- model.iou_thresh = 0.45
33
- model.view_img = False
34
- model.save_img = True
35
 
36
  pred = model.predict(source='data/images',yaml='data/coco.yaml', img_size=640)
37
  ```
 
25
  ```python
26
  from yolov6 import YOLOV6
27
 
28
+ model = YOLOV6(weights='kadirnar/yolov6s-v2.0', device='cuda:0', hf_model=True)
29
 
30
  model.classes = None
31
+ model.conf = 0.25
32
+ model.iou = 0.45
33
+ model.show = False
34
+ model.save = True
35
 
36
  pred = model.predict(source='data/images',yaml='data/coco.yaml', img_size=640)
37
  ```