TooTouch/MemSeg

IndexError: index 0 is out of bounds for axis 0 with size 0

Closed this issue · 2 comments

Do you have a great god who knows how to solve this problem?

File "F:\MemSeg-main\models\memory_module.py", line 33, in update
input_normal, _, _ = self.normal_dataset[samples_idx[i]]
~~~~~~~~~~~^^^

@A0401A

If you defined self.normal_dataset as MemSegDataset class in data folder, self.normal_dataset should return img, mask, and target as follows

return img, mask, target

Did you use your custom dataset, not MemSegDataset?

Thank you for your reply. I am using MemSegDataset, and I added a dot before the dataset path to avoid any errors.