fcakyon commited on
Commit
5233606
1 Parent(s): a4bd774

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +37 -1
README.md CHANGED
@@ -1,3 +1,39 @@
1
  ---
2
- license: agpl-3.0
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ license: gpl-3.0
3
+ inference: false
4
+ tags:
5
+ - instance-segmentation
6
+ - computer-vision
7
+ - vision
8
+ - yolo
9
+ - yolov5
10
  ---
11
+
12
+ ### How to use
13
+
14
+ - Install yolov5:
15
+
16
+ ```bash
17
+ pip install -U yolov5
18
+ ```
19
+
20
+ - Load model and perform prediction:
21
+
22
+ ```python
23
+ import yolov5
24
+
25
+ # load model
26
+ model = yolov5.load('fcakyon/yolov5n-seg-v7.0')
27
+
28
+ # set image
29
+ img = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
30
+
31
+ # perform inference
32
+ results = model(img)
33
+ ```
34
+
35
+ - Finetune the model on your custom dataset:
36
+
37
+ ```bash
38
+ yolov5 segment train --img 128 --weights fcakyon/yolov5n-seg-v7.0 --epochs 1 --device cpu
39
+ ```