diff --git a/datasets/raise/extract_4_color_channel_images_from_raw.py b/datasets/raise/extract_4_color_channel_images_from_raw.py index fbe3df4..a3a6d07 100644 --- a/datasets/raise/extract_4_color_channel_images_from_raw.py +++ b/datasets/raise/extract_4_color_channel_images_from_raw.py @@ -5,8 +5,8 @@ from utils import Color, getColorChannel IMAGE_FILE_PATH = '../rafael/240424/photos/DSC03294.ARW' Y_RANGE = 5 X_RANGE = 25 -ORIGINAL_INTERESTING_POSITION = (5752, 1695) -INTERESTING_POSITION = np.array(ORIGINAL_INTERESTING_POSITION)[::-1] // 2 +ORIGINAL_INTERESTING_POSITION = [5752, 1695][::-1] +INTERESTING_POSITION = np.array(ORIGINAL_INTERESTING_POSITION) // 2 colorChannel = getColorChannel(IMAGE_FILE_PATH, Color.RED) print(colorChannel.shape) @@ -14,8 +14,8 @@ print(colorChannel.shape) def crop(image, interestingPosition, yRange, xRange): return image[interestingPosition[0] - yRange: interestingPosition[0] + yRange, interestingPosition[1] - xRange: interestingPosition[1] + xRange] +#colorChannel = crop(multipleColorsImage, ORIGINAL_INTERESTING_POSITION, Y_RANGE, X_RANGE) colorChannel = crop(colorChannel, INTERESTING_POSITION, Y_RANGE, X_RANGE) -colorChannel = colorChannel[INTERESTING_POSITION[0] - Y_RANGE: INTERESTING_POSITION[0] + Y_RANGE, INTERESTING_POSITION[1] - X_RANGE: INTERESTING_POSITION[1] + X_RANGE] print(colorChannel.shape) plt.imshow(colorChannel) plt.show() \ No newline at end of file