

# 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


