diff --git a/dan/ocr/manager/dataset.py b/dan/ocr/manager/dataset.py index 56adf48023313fb2f62644c0ccceb7295f7658bd..6736de7b7e69df6e5c28e4f109d2a1eac489f46b 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): """