From 5c37875f6e22119056eda663666004cc04e9bdbe Mon Sep 17 00:00:00 2001 From: manonBlanco <blanco@teklia.com> Date: Tue, 22 Aug 2023 16:25:34 +0200 Subject: [PATCH] Send str to OpenCV --- dan/ocr/manager/dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dan/ocr/manager/dataset.py b/dan/ocr/manager/dataset.py index 56adf480..6736de7b 100644 --- a/dan/ocr/manager/dataset.py +++ b/dan/ocr/manager/dataset.py @@ -120,7 +120,8 @@ class OCRDataset(Dataset): if self.load_in_memory: return self.samples[i]["img"] - return self.preprocessing_transforms(read_image(self.samples[i]["path"])) + image_path = str(self.samples[i]["path"]) + return self.preprocessing_transforms(read_image(image_path)) def compute_std_mean(self): """ -- GitLab