Update model-index.yaml
Browse files- model-index.yaml +13 -14
model-index.yaml
CHANGED
@@ -1,14 +1,13 @@
|
|
1 |
-
|
2 |
-
- name: Rock vs Non-Rock
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
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
|
|