Ugness
commented
4 years ago 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.
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