a48b055a628bfa648a0b5a371838848e997440e2,unet3d/normalize.py,,get_complete_foreground,#,18

Before Change

def get_complete_foreground(subject_folders):
    for i, subject_folder in enumerate(subject_folders):
        background_path = os.path.join(subject_folder, "background.nii.gz")
        image = nib.load(background_path)
        image_foreground = image.get_data() == 0
        if i == 0:
            foreground = image_foreground
            reference_image = image
        else:
            foreground[image_foreground] = 1

    return new_img_like(reference_image, foreground)


def normalize_data(data, mean, std):
    data -= mean[:, np.newaxis, np.newaxis, np.newaxis]

After Change

        else:
            foreground[subject_foreground > 0] = 1

    return new_img_like(nib.load(training_data_files[0][-1]), foreground)


def get_foreground_from_set_of_files(set_of_files, background_value=0, tolerance=0.00001):
    for i, image_file in enumerate(set_of_files):

Instances