02_create_volumes_sex_determination.py 文件源码

def process_study(study_id, out_dir):
    isometric_volume = np.load('../data_proc/stage1/isotropic_volumes_1mm/{}.npy'.format(study_id))
    mean = np.mean(isometric_volume).astype(np.float32)
    std = np.std(isometric_volume).astype(np.float32)
    volume_resized = scipy.ndimage.interpolation.zoom(isometric_volume,
                                                      np.divide(64, isometric_volume.shape),
                                                      mode='nearest')
    volume_resized = (volume_resized.astype(np.float32) - mean) / (std + 1e-7)
    for i in range(7):
        z_shift = random.randint(0, 5)
        z0 = (volume_resized.shape[0]//2) - z_shift
        z1 = volume_resized.shape[0] - z_shift
        y_shift = random.randint(0, 5)
        y0 = y_shift
        y1 = (volume_resized.shape[1]//2) + y_shift
        volume_resized_sample = volume_resized[z0:z1, y0:y1, :]
        volume_resized_sample = np.expand_dims(volume_resized_sample, axis=3)
        out_filepath = os.path.join(out_dir, '{}.npy'.format(uuid4()))
        np.save(out_filepath, volume_resized_sample)
    return