From 0c468c18c6cb14d653864506caafb6495f7daf6d Mon Sep 17 00:00:00 2001
From: alexcbb <alexchapin@hotmail.fr>
Date: Fri, 28 Jul 2023 16:27:08 +0200
Subject: [PATCH] Fix ycb
---
osrt/data/ycbv.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/osrt/data/ycbv.py b/osrt/data/ycbv.py
index 6f07c80..1c67c7b 100644
--- a/osrt/data/ycbv.py
+++ b/osrt/data/ycbv.py
@@ -326,8 +326,8 @@ class YCBVideo2D(Dataset):
path_synth = self.path + "/train_synt/"
self.images = extract_images_path(path_real, mode, self.images)
self.images = extract_images_path(path_synth, mode, self.images)
- self.masks = extract_images_path(path_real, mode, self.masks, "masks")
- self.masks = extract_images_path(path_synth, mode, self.masks, "masks")
+ self.masks = extract_images_path(path_real, mode, self.masks, "mask")
+ self.masks = extract_images_path(path_synth, mode, self.masks, "mask")
dataset_name = 'YCB'
print(f"Load dataset {dataset_name} in mode {self.mode}")
--
GitLab