Skip to content
Snippets Groups Projects
Commit 2a70c5b5 authored by Alexandre Chapin's avatar Alexandre Chapin :race_car:
Browse files

Fix YCB dataset

parent 7b372512
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
osrt/data/ycbv.py
+ 1
− 1
View file @ 2a70c5b5
...@@ -323,7 +323,7 @@ class YCBVideo2D(Dataset): ...@@ -323,7 +323,7 @@ class YCBVideo2D(Dataset):
self.images = np.concatenate(self.images, temp) self.images = np.concatenate(self.images, temp)
else: else:
path_real = self.path + "/train_real/" 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_real, mode, self.images)
self.images = extract_images_path(path_synth, 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_real, mode, self.masks, "masks")
......
slurm/sa_train_clever.slurm → slurm/sa_train_clevr.slurm
+ 0
− 0
View file @ 2a70c5b5
File moved
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment