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