from batchgenerators.utilities.file_and_folder_operations import * | |
import numpy as np | |
if __name__ == '__main__': | |
input_file = '/home/fabian/data/nnUNet_preprocessed/Task004_Hippocampus/nnUNetPlansv2.1_plans_3D.pkl' | |
output_file = '/home/fabian/data/nnUNet_preprocessed/Task004_Hippocampus/nnUNetPlansv2.1_LISA_plans_3D.pkl' | |
a = load_pickle(input_file) | |
a['plans_per_stage'][0]['batch_size'] = int(np.floor(6 / 9 * a['plans_per_stage'][0]['batch_size'])) | |
save_pickle(a, output_file) |