Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

catch numpy warning

Browse Source
This commit is contained in:
Vladimir Mandic
2023-08-26 17:19:35 +02:00
parent 91406bc9a0
commit 0f10a9ce63
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing.py
+3 -2
View File
@@ -583,14 +583,15 @@ def validate_sample(sample):
try:
sample = sample.astype(np.uint8)
return sample
except Exception as e:
except (Warning, Exception) as e:
shared.log.error(f'Failed to validate sample values: {e}')
ok = False
if not ok:
try:
sample = np.nan_to_num(sample, nan=0, posinf=255, neginf=0)
sample = sample.astype(np.uint8)
except Exception as e:
shared.log.debug('Corrected sample values')
except (Warning, Exception) as e:
shared.log.error(f'Failed to correct sample values: {e}')
sample = np.zeros_like(sample)
sample = sample.astype(np.uint8)