diff --git a/config/cityscapes.yml b/config/cityscapes.yml new file mode 100644 index 0000000..861ad4c --- /dev/null +++ b/config/cityscapes.yml @@ -0,0 +1,48 @@ +batch_size: 4 +iters: 80000 + +train_dataset: + type: Dataset + dataset_root: data/cricket + train_path: data/cricket/train.txt + num_classes: 9 + transforms: + - type: ResizeStepScaling + min_scale_factor: 0.5 + max_scale_factor: 2.0 + scale_step_size: 0.25 + - type: RandomPaddingCrop + crop_size: [1024, 512] + - type: RandomHorizontalFlip + - type: RandomDistort + brightness_range: 0.4 + contrast_range: 0.4 + saturation_range: 0.4 + - type: Normalize + mode: train + +val_dataset: + type: Dataset + dataset_root: data/cricket + val_path: data/cricket/val.txt + num_classes: 9 + transforms: + - type: Normalize + mode: val + + +optimizer: + type: sgd + momentum: 0.9 + weight_decay: 4.0e-5 + +lr_scheduler: + type: PolynomialDecay + learning_rate: 0.0025 + end_lr: 0 + power: 0.9 + +loss: + types: + - type: CrossEntropyLoss + coef: [1]