Skip to content
Snippets Groups Projects

Store absolute path of images

Merged Manon Blanco requested to merge format-absolute-path into main
1 file
+ 6
4
Compare changes
  • Side-by-side
  • Inline
@@ -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)
Loading