Mask r-cnn 无法训练的问题

Question

在做 https://github.com/matterport/Mask_RCNN 的复现。
在复现train_shpes时，在heads层训练时，卡在了Epoch 1/1。我观察下gpu和cpu，都没有工作
我在停止代码运行时发现停在了

File "", line 1, in
runfile('D:/py/Mask_RCNN-master/samples/shapes/train_shapes.py', wdir='D:/py/Mask_RCNN-master/samples/shapes')

File "D:\anaconda\envs\tensorflow\lib\site-packages\spyder\utils\site\sitecustomize.py", line 705, in runfile
execfile(filename, namespace)

File "D:\anaconda\envs\tensorflow\lib\site-packages\spyder\utils\site\sitecustomize.py", line 102, in execfile
exec(compile(f.read(), filename, 'exec'), namespace)

File "D:/py/Mask_RCNN-master/samples/shapes/train_shapes.py", line 258, in
layers='heads')

File "D:\py\Mask_RCNN-master\mrcnn\model.py", line 2352, in train
use_multiprocessing=True,

File "D:\anaconda\envs\tensorflow\lib\site-packages\keras\legacy\interfaces.py", line 87, in wrapper
return func(*args, **kwargs)

File "D:\anaconda\envs\tensorflow\lib\site-packages\keras\engine\training.py", line 2011, in fit_generator
generator_output = next(output_generator)

File "D:\anaconda\envs\tensorflow\lib\site-packages\keras\utils\data_utils.py", line 644, in get
time.sleep(self.wait_time)
有大佬知道怎么解决吗，或者有谁出现了相同的问题吗？？

Answer 1

问题总算解决了。细看了一下代码，keras在windows下进行多线程处理的时，会有问题。我在查看model.py中train函数时，发现作者已经注释了
Keras fails on Windows when using multiprocessing workers.
这是在windows系统下会出现的问题，虽然我的程序没有报错。
之前因为将keras升级到2.2.0，导致keras版本过高出现了一些问题，按之前别人的方法偷懒将keras降到了2.0.8.
具体问题https://blog.csdn.net/u014513323/article/details/80659352
后来我将keras升级到2.1.6，程序挂起的问题就解决了。
如果你想用keras2.2.0版 只需要将model.py中 topology替换成saving就行了。