From fb23b4d81a88fe9d93120bf0ca226da98eb4e2d9 Mon Sep 17 00:00:00 2001 From: manonBlanco <blanco@teklia.com> Date: Thu, 6 Jul 2023 13:53:38 +0200 Subject: [PATCH] Nits --- dan/datasets/format/atr.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/dan/datasets/format/atr.py b/dan/datasets/format/atr.py index 594d9e44..66e38e1f 100644 --- a/dan/datasets/format/atr.py +++ b/dan/datasets/format/atr.py @@ -63,11 +63,13 @@ 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}", - ).realpath(), + "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