'shuffle': True, 'num_workers': PARAMS.num_workers, 'pin_memory': PARAMS.pin_memory }, 'inference': { 'batch_size': PARAMS.batch_size_inference, 'shuffle': False, 'num_workers': PARAMS.num_workers, 'pin_memory': PARAMS.pin_memory }, }, 'augmentation_params': { 'image_augment_train': aug.intensity_seq, 'image_augment_with_target_train': aug.resize_seq(resize_target_size=PARAMS.resize_target_size), 'image_augment_inference': aug.resize_to_fit_net( resize_target_size=PARAMS.resize_target_size), 'image_augment_with_target_inference': aug.resize_to_fit_net( resize_target_size=PARAMS.resize_target_size) }, }, 'resize_tta': { 'dataset_params': { 'h': PARAMS.image_h, 'w': PARAMS.image_w, 'image_source': PARAMS.image_source, 'target_format': PARAMS.target_format, 'MEAN': MEAN,
'STD': STD }, 'loader_params': {'training': {'batch_size': PARAMS.batch_size_train, 'shuffle': True, 'num_workers': PARAMS.num_workers, 'pin_memory': PARAMS.pin_memory }, 'inference': {'batch_size': PARAMS.batch_size_inference, 'shuffle': False, 'num_workers': PARAMS.num_workers, 'pin_memory': PARAMS.pin_memory }, }, 'augmentation_params': {'image_augment_train': aug.intensity_seq, 'image_augment_with_target_train': aug.resize_seq( resize_target_size=PARAMS.resize_target_size), 'image_augment_inference': aug.resize_to_fit_net( resize_target_size=PARAMS.resize_target_size), 'image_augment_with_target_inference': aug.resize_to_fit_net( resize_target_size=PARAMS.resize_target_size) }, }, 'resize_tta': {'dataset_params': {'h': PARAMS.image_h, 'w': PARAMS.image_w, 'image_source': PARAMS.image_source, 'use_depth': USE_DEPTH, 'MEAN': MEAN, 'STD': STD }, 'loader_params': {'training': {'batch_size': PARAMS.batch_size_train, 'shuffle': True,