aiweld commited on
Commit
d832f7c
·
verified ·
1 Parent(s): ed1511b

Rename model.yaml to config.json

Browse files
Files changed (2) hide show
  1. config.json +22 -0
  2. model.yaml +0 -26
config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "aiweld-annotate-yolov8-seg",
3
+ "type": "detector",
4
+ "framework": "ultralytics",
5
+ "task": "segmentation",
6
+ "weights": {
7
+ "url": "best.pt"
8
+ },
9
+ "labels": [
10
+ "Porosity",
11
+ "Slag Inclusion",
12
+ "Incomplete Penetration",
13
+ "Incomplete Fusion",
14
+ "Wormhole",
15
+ "Crack",
16
+ "IQI",
17
+ "Norm",
18
+ "Linear Porosity",
19
+ "Artifact",
20
+ "Weld"
21
+ ]
22
+ }
model.yaml DELETED
@@ -1,26 +0,0 @@
1
- # YOLOv8 Segmentation model configuration for CVAT
2
-
3
- # Task type
4
- task: segment
5
- framework: ultralytics
6
- format: yolov8
7
-
8
- # Number of classes
9
- nc: 11
10
-
11
- # Class names
12
- names:
13
- 0: Porosity
14
- 1: Slag Inclusion
15
- 2: Incomplete Penetration
16
- 3: Incomplete Fusion
17
- 4: Wormhole
18
- 5: Crack
19
- 6: IQI
20
- 7: Norm
21
- 8: Linear Porosity
22
- 9: Artifact
23
- 10: Weld
24
-
25
- # Model weights file (must exist in repo)
26
- weights: best.pt