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

Replace tabs with spaces in datasets/raise/extract_noise.py

Browse Source
This commit is contained in:
Benjamin Loison 2024-04-05 23:57:37 +02:00
parent ba452096d5
commit a83e8bb1e0
No known key found for this signature in database
1 changed files with 22 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
datasets/raise/extract_noise.py
View File

@ -21,29 +21,29 @@ imagesFileNames = os.listdir(imagesFolderPath)
pbar = tqdm(total = len(imagesFileNames))
def treatImage(imageFileName):
global mean, numberOfImagesInMean, pbar
imageFilePath = f'{imagesFolderPath}/{imageFileName}'
imagePil = Image.open(imageFilePath)
imageNpArray = img_as_float(np.array(imagePil))
print('Before mean computation')
imageNoiseNpArray = imageNpArray - denoise_tv_chambolle(imageNpArray, weight=0.2, channel_axis=-1)
print('After mean computation')
with mutex:
print('Start mutex section')
if mean is None:
print('Inter A')
mean = imageNoiseNpArray
else:
print('Inter B')
mean = ((mean * numberOfImagesInMean) + imageNoiseNpArray) / (numberOfImagesInMean + 1)
print('Intermediary 0')
numberOfImagesInMean += 1
print('Intermediary 1')
pbar.update(numberOfImagesInMean)
print('End mutex section')
global mean, numberOfImagesInMean, pbar
imageFilePath = f'{imagesFolderPath}/{imageFileName}'
imagePil = Image.open(imageFilePath)
imageNpArray = img_as_float(np.array(imagePil))
print('Before mean computation')
imageNoiseNpArray = imageNpArray - denoise_tv_chambolle(imageNpArray, weight=0.2, channel_axis=-1)
print('After mean computation')
with mutex:
print('Start mutex section')
if mean is None:
print('Inter A')
mean = imageNoiseNpArray
else:
print('Inter B')
mean = ((mean * numberOfImagesInMean) + imageNoiseNpArray) / (numberOfImagesInMean + 1)
print('Intermediary 0')
numberOfImagesInMean += 1
print('Intermediary 1')
pbar.update(numberOfImagesInMean)
print('End mutex section')
with Pool(5) as p:
p.map(treatImage, imagesFileNames)
p.map(treatImage, imagesFileNames)
with open(npArrayFilePath, 'wb') as f:
np.save(f, mean)
np.save(f, mean)