keffy commited on
Commit
786d624
·
verified ·
1 Parent(s): f064922

Update model-index.yaml

Browse files
Files changed (1) hide show
  1. model-index.yaml +13 -14
model-index.yaml CHANGED
@@ -1,14 +1,13 @@
1
- model_index:
2
- - name: Rock vs Non-Rock YOLOv8
3
- metadata:
4
- library_name: ultralytics
5
- framework: pytorch
6
- tags:
7
- - yolov8
8
- - object-detection
9
- - rock-classification
10
- config: config.yaml
11
- weights: best.pt
12
- labels:
13
- 0: rock
14
- 1: non-rock
 
1
+ model-index:
2
+ - name: Rock vs Non-Rock Model
3
+ task:
4
+ type: object-detection
5
+ framework: pytorch
6
+ tags:
7
+ - yolov8
8
+ - rock-classification
9
+ config: hvc_model/config.yaml
10
+ weights: hvc_model/best.pt
11
+ labels:
12
+ 0: rock
13
+ 1: non-rock