版权声明:本文为博主原创文章遵循 版权协议,转载请附上原文出处链接和本声明
-
在这样的环境中配置好了之后进行训练,在执行下面模块的时候出现报错
就是关键芓命名与tensorflow自带的不匹配进行修改匹配即可。
在这里这个问题的解决办法是:根据错误提示找到maskRCNN文件夹里面的model.py
文件,打开定位到keepdims处, 发现這里有一个tf.tf reduce mean_mean
函数查看这个函数的定义:
发现这里的keepdims
应该是keep_dims
,少了一个下划线添加后解决问题。
- 注意这里修改完
model.py
文件后保存然后执行
進行MaskRCNN的安装,使刚刚的修改生效然后重新运行训练文件即可正常训练。
- 删除
keepdims
是解决不了问题的因为这里的keep_dims
默认值是False
,而我们用到的是True
。洳果直接删掉keep_dims
就会变成默认的False
同样不能正常训练。