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

Fix ycb

parent 2a70c5b5
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
osrt/data/ycbv.py
+ 2
− 2
View file @ 0c468c18
......@@ -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}")
......
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