| 12345678910111213141516171819202122232425262728293031323334 |
- # Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
- dataset_path: F:\DevTools\datasets\renyaun\1012\spilt
- #train: images/train # train images (relative to 'path') 128 images
- #val: images/train # val images (relative to 'path') 128 images
- #test: images/test # test images (optional)
- #train parameters
- num_classes: 5
- opt: 'adamw'
- batch_size: 2
- epochs: 10
- lr: 0.005
- momentum: 0.9
- weight_decay: 0.0001
- lr_step_size: 3
- lr_gamma: 0.1
- num_workers: 4
- print_freq: 10
- target_type: polygon
- enable_logs: True
- augmentation: True
- ## Classes
- #names:
- # 0: fire
- # 1: dust
- # 2: move_machine
- # 3: open_machine
- # 4: close_machine
|
