Update config.yaml
Browse files- config.yaml +12 -3
config.yaml
CHANGED
|
@@ -1,7 +1,14 @@
|
|
| 1 |
-
|
| 2 |
-
train: images/train
|
| 3 |
-
val: images/val
|
| 4 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
names:
|
| 6 |
0: Porosity
|
| 7 |
1: Slag Inclusion
|
|
@@ -15,3 +22,5 @@ names:
|
|
| 15 |
9: Artifact
|
| 16 |
10: Weld
|
| 17 |
|
|
|
|
|
|
|
|
|
| 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
|
|
|
|
| 22 |
9: Artifact
|
| 23 |
10: Weld
|
| 24 |
|
| 25 |
+
# Model weights file (must exist in repo)
|
| 26 |
+
weights: best.pt
|