From f73ffbc11db56091a48a2175a7daaefc11b50aa4 Mon Sep 17 00:00:00 2001
From: Manon blanco <blanco@teklia.com>
Date: Fri, 7 Jul 2023 07:43:30 +0000
Subject: [PATCH] Store absolute path of images
---
dan/datasets/format/atr.py | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/dan/datasets/format/atr.py b/dan/datasets/format/atr.py
index 7da5a8da..66e38e1f 100644
--- a/dan/datasets/format/atr.py
+++ b/dan/datasets/format/atr.py
@@ -63,10 +63,12 @@ class ATRDatasetFormatter:
def parse_labels(self, set_name, file_name):
return {
- "img_path": os.path.join(
- self.image_folder,
- set_name,
- f"{os.path.splitext(file_name)[0]}.{self.image_format}",
+ "img_path": os.path.realpath(
+ os.path.join(
+ self.image_folder,
+ set_name,
+ f"{os.path.splitext(file_name)[0]}.{self.image_format}",
+ )
),
"label": self.read_file(
os.path.join(self.labels_folder, set_name, file_name)
--
GitLab