模型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  半夜打老虎  阅读(2030)  评论(0编辑  收藏  举报