Create config.yaml
Browse files- config.yaml +3 -0
config.yaml
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
# YOLOv8 model configuration for Hugging Face
|
2 |
+
nc: 2 # Number of classes (rock and non-rock)
|
3 |
+
names: ['rock', 'non-rock'] # Class names
|