bmombie commited on
Commit
4db8b76
1 Parent(s): 27abb8f

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +7 -6
config.yaml CHANGED
@@ -1,6 +1,7 @@
1
- path: /home/sagemaker-user/
2
- train: obj_train_data
3
- val: obj_train_data
4
-
5
- names:
6
- 0: bicycle_rider
 
 
1
+ model_type: yolov8
2
+ model_architecture: yolov8n
3
+ input_size: 640
4
+ num_classes: 1
5
+ classes:
6
+ 0: bicycle_rider
7
+ pipeline_tag: object-detection