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

一、问题：

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

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

  二、解决方法：

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

?
1
2
3
4
5
6
7
8
9
10
11
12
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 @   半夜打老虎  阅读(2052)  评论(0)  编辑  收藏  举报
编辑推荐：
· AI与.NET技术实操系列：基于图像分类模型对图像进行分类
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语，封装的思维：从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热？
阅读排行：
· 25岁的心里话
· 闲置电脑爆改个人服务器（超详细） #公网映射 #Vmware虚拟网络编辑器
· 零经验选手，Compose 一天开发一款小游戏！
· 因为Apifox不支持离线，我果断选择了Apipost！
· 通过 API 将Deepseek响应流式内容输出到前端
点击右上角即可分享
微信分享提示
AI IDE Trae