Object Detection
computer-vision
yolov6
pypi
kadirnar commited on
Commit
dad1dff
1 Parent(s): f17d9c2

Upload 2 files

Browse files
Files changed (2) hide show
  1. README.md +28 -0
  2. yolov6s.pt +3 -0
README.md CHANGED
@@ -1,3 +1,31 @@
1
  ---
2
  license: gpl-3.0
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: gpl-3.0
3
+ inference: false
4
+ tags:
5
+ - object-detection
6
+ - computer-vision
7
+ - yolov6
8
+ - pypi
9
+ datasets:
10
+ - detection-datasets/coco
11
  ---
12
+
13
+ ### Installation
14
+ ```
15
+ pip install yolov6detect
16
+ ```
17
+
18
+ ### Yolov6 Inference
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
26
+ model.iou_thresh = 0.45
27
+ model.view_img = False
28
+ model.save_img = True
29
+
30
+ pred = model.predict(source='data/images',yaml='data/coco.yaml', img_size=640)
31
+ ```
yolov6s.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8e572fe915ca45cc8c366ef5d696e660b69f11e0a71924c0e9ba57cb87314671
3
+ size 38101272