模型Mask_Rcnn训练错误：RuntimeError: It looks like you are subclassing 'Model'

一、问题：

         最近在keras下利用两个GPU训练Mask_Rcnn模型的时候出现以下错误。

error:RuntimeError: It looks like you are subclassing 'Model'	
and you forgot to call 'super(YourClass, self).__init__()'

  二、解决方法：

         在mrcnn/parallel_model.py文件中找到class ParallelModel(KM.Model),在类里面添加super(ParallelModel, self).__init__()

class ParallelModel(KM.Model):
    def __init__(self, keras_model, gpu_count):
    """
　　 Class constructor.
       keras_model: The Keras model to parallelize
       gpu_count: Number of GPUs. Must be > 1
　　 """
　　 super(ParallelModel, self).__init__()
　　 self.inner_model = keras_model
　　 self.gpu_count = gpu_count
　　 merged_outputs = self.make_parallel()
　　 super(ParallelModel, self).__init__(inputs=self.inner_model.inputs, outputs=merged_outputs)

　　

 

posted @ 2020-03-20 18:06  半夜打老虎  阅读(2066)  评论(0)    收藏  举报