From 259df647a38ce6b6c03805d2ed6edcb0f0ed81ad Mon Sep 17 00:00:00 2001
From: manonBlanco <blanco@teklia.com>
Date: Fri, 3 Nov 2023 14:57:54 +0100
Subject: [PATCH] Add early return
---
dan/ocr/predict/inference.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/dan/ocr/predict/inference.py b/dan/ocr/predict/inference.py
index e690b72d..c869d4e1 100644
--- a/dan/ocr/predict/inference.py
+++ b/dan/ocr/predict/inference.py
@@ -125,6 +125,9 @@ class DAN:
image = read_image(path)
preprocessed_image = self.preprocessing_transforms(image)
+ if self.mean is None and self.std is None:
+ return preprocessed_image, preprocessed_image
+
size = preprocessed_image.shape
normalized_image = torch.zeros(size)
--
GitLab