tensorflow配置Mask-RCNN报错总结

1、在环境中配置好了之后进行训练,在执行下面模块的时候出现报错

loss = (
                tf.reduce_mean(layer.output, keep_dims=True)
                * self.config.LOSS_WEIGHTS.get(name, 1.))
            self.keras_model.add_loss(loss)

 错误信息:

TypeError: reduce_mean() got an unexpected keyword argument 'keepdims'

这种典型问题:...got an unexpected keyword argument...
就是关键字命名与tensorflow自带的不匹配,这是由于tensorflow版本问题引起的,对keepdims进行修改为keep_dims。

2、每次修改mrcnn中文件后都要进行编译

python setup.py install


 

猜你喜欢

转载自blog.csdn.net/qq_30159015/article/details/82108699