diff --git a/aidge_export_cpp/export_utils.py b/aidge_export_cpp/export_utils.py index 7830800803c357150f84444e508a5e261c798166..4cbc38d7d43d56893dd093455310f891226023d6 100644 --- a/aidge_export_cpp/export_utils.py +++ b/aidge_export_cpp/export_utils.py @@ -215,6 +215,8 @@ def normalize(array): """ Normalize an input image between -1 and 1 """ + if array.max() == array.min(): + return array/array.max() array = (array - array.min()) / (array.max() - array.min()) return 2 * array - 1