Benjamin_Loison/Robust_image_source_identification_on_modern_smartphones
1
0
Fork 0
Code Issues 69 Pull Requests Packages Projects Releases 1 Wiki Activity

Add extract_4_color_channel_images_from_raw.py

Browse Source
This commit is contained in:
Benjamin Loison
2024-05-23 12:21:04 +02:00
parent 7db83a10c9
commit 24b6971632
1 changed files with 21 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
datasets/raise/extract_4_color_channel_images_from_raw.py Normal file
View File

@@ -0,0 +1,21 @@
import numpy as np
import matplotlib.pyplot as plt
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
colorChannel = getColorChannel(IMAGE_FILE_PATH, Color.RED)
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(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()