From 2a70c5b5fe17373b651e0fde370a7663aaecd2a6 Mon Sep 17 00:00:00 2001
From: alexcbb <alexchapin@hotmail.fr>
Date: Fri, 28 Jul 2023 15:58:37 +0200
Subject: [PATCH] Fix YCB dataset
---
osrt/data/ycbv.py | 2 +-
slurm/{sa_train_clever.slurm => sa_train_clevr.slurm} | 0
2 files changed, 1 insertion(+), 1 deletion(-)
rename slurm/{sa_train_clever.slurm => sa_train_clevr.slurm} (100%)
diff --git a/osrt/data/ycbv.py b/osrt/data/ycbv.py
index d9865ce..6f07c80 100644
--- a/osrt/data/ycbv.py
+++ b/osrt/data/ycbv.py
@@ -323,7 +323,7 @@ class YCBVideo2D(Dataset):
self.images = np.concatenate(self.images, temp)
else:
path_real = self.path + "/train_real/"
- path_synth = self.path + "/train_synth/"
+ 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")
diff --git a/slurm/sa_train_clever.slurm b/slurm/sa_train_clevr.slurm
similarity index 100%
rename from slurm/sa_train_clever.slurm
rename to slurm/sa_train_clevr.slurm
--
GitLab