74 lines
1.4 KiB
YAML
74 lines
1.4 KiB
YAML
dataset_params:
|
|
root_dir: ../TED384-v2
|
|
frame_shape: null
|
|
id_sampling: True
|
|
augmentation_params:
|
|
flip_param:
|
|
horizontal_flip: True
|
|
time_flip: True
|
|
jitter_param:
|
|
brightness: 0.1
|
|
contrast: 0.1
|
|
saturation: 0.1
|
|
hue: 0.1
|
|
|
|
model_params:
|
|
common_params:
|
|
num_tps: 10
|
|
num_channels: 3
|
|
bg: True
|
|
multi_mask: True
|
|
generator_params:
|
|
block_expansion: 64
|
|
max_features: 512
|
|
num_down_blocks: 3
|
|
dense_motion_params:
|
|
block_expansion: 64
|
|
max_features: 1024
|
|
num_blocks: 5
|
|
scale_factor: 0.25
|
|
avd_network_params:
|
|
id_bottle_size: 128
|
|
pose_bottle_size: 128
|
|
|
|
|
|
train_params:
|
|
num_epochs: 100
|
|
num_repeats: 150
|
|
epoch_milestones: [70, 90]
|
|
lr_generator: 2.0e-4
|
|
batch_size: 12
|
|
scales: [1, 0.5, 0.25, 0.125]
|
|
dataloader_workers: 6
|
|
checkpoint_freq: 50
|
|
dropout_epoch: 35
|
|
dropout_maxp: 0.5
|
|
dropout_startp: 0.0
|
|
dropout_inc_epoch: 10
|
|
bg_start: 0
|
|
transform_params:
|
|
sigma_affine: 0.05
|
|
sigma_tps: 0.005
|
|
points_tps: 5
|
|
loss_weights:
|
|
perceptual: [10, 10, 10, 10, 10]
|
|
equivariance_value: 10
|
|
warp_loss: 10
|
|
bg: 10
|
|
|
|
train_avd_params:
|
|
num_epochs: 30
|
|
num_repeats: 500
|
|
batch_size: 256
|
|
dataloader_workers: 24
|
|
checkpoint_freq: 10
|
|
epoch_milestones: [20, 25]
|
|
lr: 1.0e-3
|
|
lambda_shift: 1
|
|
random_scale: 0.25
|
|
|
|
visualizer_params:
|
|
kp_size: 5
|
|
draw_border: True
|
|
colormap: 'gist_rainbow'
|