cvlab-yonsei / MNAD

An official implementation of "Learning Memory-guided Normality for Anomaly Detection" (CVPR 2020) in PyTorch.
333 stars 82 forks source link

An error occurred while training the Shanghai dataset #31

Open xiximin opened 3 years ago

xiximin commented 3 years ago

Traceback (most recent call last): File "Train.py", line 154, in for j,(imgs) in enumerate(train_batch): File "/root/userfolder/software/anaconda3/envs/MemG/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 819, in next return self._process_data(data) File "/root/userfolder/software/anaconda3/envs/MemG/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 846, in _process_data data.reraise() File "/root/userfolder/software/anaconda3/envs/MemG/lib/python3.6/site-packages/torch/_utils.py", line 369, in reraise raise self.exc_type(msg) IndexError: Caught IndexError in DataLoader worker process 1. Original Traceback (most recent call last): File "/root/userfolder/software/anaconda3/envs/MemG/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop data = fetcher.fetch(index) File "/root/userfolder/software/anaconda3/envs/MemG/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/root/userfolder/software/anaconda3/envs/MemG/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/root/userfolder/code/projects1/MNAD/FFPMem/model/utils.py", line 71, in getitem image = np_load_frame(self.videos[video_name]['frame'][frame_name+i], self._resize_height, self._resize_width) IndexError: list index out of range

Yuan98Yu commented 3 years ago

find this line(line 62) in file 'model/utils.py' origin: for i in range(len(self.videos[video_name]['frame'])-self._time_step): modified: for i in range(len(self.videos[video_name]['frame'])-self._time_step-self._num_pred):

renqi1120 commented 3 years ago

you need modify shanghai's labels file 《frame_labels_shanghai.npy》,Incorrect label format.

huyi1998 commented 2 years ago

you need modify shanghai's labels file 《frame_labels_shanghai.npy》,Incorrect label format.

可以看看你的代码吗,我不知道为啥,train老是遇到各种错误

Celinewxy commented 4 months ago

回溯(最近一次调用最后一次): 文件“Train.py”,第 154 行, for j,(imgs) in enumerate(train_batch): 文件“/root/userfolder/software/anaconda3/envs/MemG/lib/python3. 6/site-packages/torch/utils/data/dataloader.py”,第 819 行,在下一个 返回 self._process_data(data) 文件“/root/userfolder/software/anaconda3/envs/MemG/lib/python3.6/ site-packages/torch/utils/data/dataloader.py”,第 846 行,在 _process_data data.reraise() 文件“/root/userfolder/software/anaconda3/envs/MemG/lib/python3.6/site-packages/ torch/_utils.py”,第 369 行,在重新引发中 raise self.exc_type(msg) IndexError:在 DataLoader 工作进程 1 中捕获 IndexError。 原始回溯(最近一次调用): 文件“/root/userfolder/software/anaconda3/envs /MemG/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py”,第 178 行,在 _worker_loop data = fetcher.fetch(index) 文件“/root/userfolder/software/anaconda3/ envs/MemG/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py​​”,第 44 行,在获取 数据 = [self.dataset[idx] for idx in possible_batched_index] 文件“/root /userfolder/software/anaconda3/envs/MemG/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py​​”,第 44 行,在 data = [self.dataset[idx] for idx in possible_batched_index] 文件“/root/userfolder/code/projects1/MNAD/FFPMem/model/utils.py”,第 71 行,在 getitem image = np_load_frame(self.videos[video_name]['frame'][frame_name+i], self._resize_height, self._resize_width) IndexError: 列表索引超出范围

请问你解决这个问题了么