TobyPDE / FRRN

Full Resolution Residual Networks for Semantic Image Segmentation
MIT License
278 stars 93 forks source link

predict error: IndexError #38

Closed dingbaojin closed 6 years ago

dingbaojin commented 6 years ago

Hello! Recently I encountered a problem when I ran your Networks.
When I ran the predict.py, it came out as follow: Traceback (most recent call last): File "/home/dingbaojin/anaconda2/lib/python2.7/multiprocessing/process.py", line 267, in _bootstrap self.run() File "/home/dingbaojin/anaconda2/lib/python2.7/multiprocessing/process.py", line 114, in run self._target(*self._args, **self._kwargs) File "/home/dingbaojin/deep_learning/Full_Resolution_Residual_Networks_test/FRRN-master/dltools/data.py", line 70, in init self.start_loading() File "/home/dingbaojin/deep_learning/Full_Resolution_Residual_Networks_test/FRRN-master/dltools/data.py", line 78, in start_loading images, targets = self.load_batch() File "/home/dingbaojin/deep_learning/Full_Resolution_Residual_Networks_test/FRRN-master/dltools/data.py", line 91, in load_batch targets = [self._convert_target(d[1]) for d in data] File "/home/dingbaojin/deep_learning/Full_Resolution_Residual_Networks_test/FRRN-master/dltools/data.py", line 125, in _convert_target selects = class_matrix[flat_targets] IndexError: index 70 is out of bounds for axis 0 with size 20 I hope your help, Thanks a lot!