diff --git a/datasets/raise/merge_single_color_channel_images_according_to_bayer_filter.py b/datasets/raise/merge_single_color_channel_images_according_to_bayer_filter.py index a11503f..01bb688 100755 --- a/datasets/raise/merge_single_color_channel_images_according_to_bayer_filter.py +++ b/datasets/raise/merge_single_color_channel_images_according_to_bayer_filter.py @@ -13,6 +13,6 @@ def getImageByColor(color): return image singleColorChannelImages = {color: getImageByColor(color) for color in Color} -multipleColorsImage = mergeSingleColorChannelImagesAccordingToBayerFilter(singleColorChannelImages, progress = True) +multipleColorsImage = mergeSingleColorChannelImagesAccordingToBayerFilter(singleColorChannelImages) plt.imsave(PREFIX + 'multiple_colors.png', multipleColorsImage) \ No newline at end of file diff --git a/datasets/raise/utils.py b/datasets/raise/utils.py index 7ca51a6..704411e 100644 --- a/datasets/raise/utils.py +++ b/datasets/raise/utils.py @@ -97,7 +97,7 @@ def getNewIndex(index, offset): def silentTqdm(data, desc = None): return data -def mergeSingleColorChannelImagesAccordingToBayerFilter(singleColorChannelImages, progress = False): +def mergeSingleColorChannelImagesAccordingToBayerFilter(singleColorChannelImages): colorImage = singleColorChannelImages[list(Color)[0]] multipleColorsImage = np.empty([dimension * 2 for dimension in colorImage.shape], dtype = np.float64) '''