def get_crop_shape(dataset_name):
    """Returns cropping shape corresponding to dataset_name."""
    if 'MPISintel' in dataset_name:
        return (384, 768)
    elif 'KITTI' in dataset_name:
        return (320, 896)
    else:
        raise ValueError('Unknown dataset name {}'.format(dataset_name))