diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py index 3aa552a..1fb0cb1 100755 --- a/datasets/raise/extract_noise.py +++ b/datasets/raise/extract_noise.py @@ -37,14 +37,16 @@ def treatImage(imageFileName): global mean, numberOfImagesInMean imageFilePath = f'{imagesFolderPath}/{imageFileName}' if imageFileName.endswith('.NEF'): - imageFileName = 'r01b6aed6t.NEF' + imageFileName = 'ra2c888f8t.NEF' imageFilePath = f'{imagesFolderPath}/{imageFileName}' + print(imageFilePath) with rawpy.imread(imageFilePath) as raw: colorDesc = raw.color_desc.decode('ascii') assert colorDesc == 'RGBG' + assert np.array_equal(raw.raw_pattern, np.array([[3, 2], [0, 1]], dtype = np.uint8)) rawImageVisible = raw.raw_image_visible.copy() - centerY = 628 - centerX = 3_014 + centerY = 2263 + centerX = 237 RANGE = 5 rawImageVisible = rawImageVisible.astype(np.float64) rawImageVisibleShape = rawImageVisible.shape