From 07dd7746ec4ef2cdd993c5e8c7f36302fe633217 Mon Sep 17 00:00:00 2001
From: Axel Farrugia <axel.farrugia@cea.fr>
Date: Fri, 2 May 2025 13:47:16 +0200
Subject: [PATCH] [Fix] The normalize function was dividing by 0 when min ==
 max

---
 aidge_export_cpp/export_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/aidge_export_cpp/export_utils.py b/aidge_export_cpp/export_utils.py
index 7830800..4cbc38d 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
 
-- 
GitLab