A ERROR

[hyk745]

new ERROR，WHY？？？

Downloading: "https://download.pytorch.org/models/vgg16-397923af.pth" to C:\Users\Administrator/.torch\models\vgg16-397923af.pth 553433881it [20:34, 448334.95it/s]

Epoch : 0 0%| | 0/24 [00:00<?, ?it/s]T raceback (most recent call last): File "E:\PytorchProject\PiCANet-Implementation-master\train.py", line 97, in for i, batch in enumerate(tqdm(dataloader)): File "d:\ProgramData\Anaconda3\lib\site-packages\tqdm_tqdm.py", line 979, in iter for obj in iterable: File "d:\ProgramData\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 615, in next batch = self.collate_fn([self.dataset[i] for i in indices]) File "d:\ProgramData\Anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 615, in batch = self.collate_fn([self.dataset[i] for i in indices]) File "E:\PytorchProject\PiCANet-Implementation-master\dataset.py", line 137, in getitem mask_name = os.path.join(self.root_dir, 'masks', self.mask_list[item]) IndexError: list index out of range

[Ugness]

I think you do not have mask images under "[ROOT_DIR]/masks".

• Please follow https://github.com/Ugness/PiCANet-Implementation#dataset if you were going to train the model.
• if you want to inference, please follow the execution guide for inference.