Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
GoooHi authored May 4, 2023
1 parent ce735f5 commit f902bf1
Showing 1 changed file with 48 additions and 0 deletions.
48 changes: 48 additions & 0 deletions config/cityscapes.yml
Original file line number Diff line number Diff line change
@@ -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]

0 comments on commit f902bf1

Please sign in to comment.