longqi-s / keras_cpn Goto Github PK

I reported the following error after deployment. Do you have any Suggestions to solve it？

Traceback (most recent call last):
File "/home/basic/workspace/Scheaven/05_cpn/keras_cpn/preprocessing/generator.py", line 28, in data_generator
imgs = data_res[0]
TypeError: 'NoneType' object is not subscriptable
Traceback (most recent call last):
File "train.py", line 69, in
layers='all')
File "/home/basic/workspace/Scheaven/05_cpn/keras_cpn/models/cpn.py", line 368, in train
use_multiprocessing=False,
File "/home/basic/anaconda3/envs/py3_keras_cpn/lib/python3.6/site-packages/keras/legacy/interfaces.py", line 91, in wrapper
return func(*args, **kwargs)
File "/home/basic/anaconda3/envs/py3_keras_cpn/lib/python3.6/site-packages/keras/engine/training.py", line 1418, in fit_generator
initial_epoch=initial_epoch)
File "/home/basic/anaconda3/envs/py3_keras_cpn/lib/python3.6/site-packages/keras/engine/training_generator.py", line 181, in fit_generator
generator_output = next(output_generator)
File "/home/basic/anaconda3/envs/py3_keras_cpn/lib/python3.6/site-packages/keras/utils/data_utils.py", line 709, in get
six.reraise(*sys.exc_info())
File "/home/basic/anaconda3/envs/py3_keras_cpn/lib/python3.6/site-packages/six.py", line 693, in reraise
raise value
File "/home/basic/anaconda3/envs/py3_keras_cpn/lib/python3.6/site-packages/keras/utils/data_utils.py", line 685, in get
inputs = self.queue.get(block=True).get()
File "/home/basic/anaconda3/envs/py3_keras_cpn/lib/python3.6/multiprocessing/pool.py", line 644, in get
raise self._value
File "/home/basic/anaconda3/envs/py3_keras_cpn/lib/python3.6/multiprocessing/pool.py", line 119, in worker
result = (True, func(*args, **kwds))
File "/home/basic/anaconda3/envs/py3_keras_cpn/lib/python3.6/site-packages/keras/utils/data_utils.py", line 626, in next_sample
return six.next(_SHARED_SEQUENCES[uid])
File "/home/basic/workspace/Scheaven/05_cpn/keras_cpn/preprocessing/generator.py", line 28, in data_generator
imgs = data_res[0]
TypeError: 'NoneType' object is not subscriptable

For others wondering, this repo uses TF1. I got it running using tf version 1.15.