当我尝试在Google colab上运行Mask_RCNN / samples / demo.ipynb时,由于出现以下错误,因此无法运行函数model = modellib.MaskRCNN(mode="inference", model_dir=MODEL_DIR, config=config)
。
AttributeError Traceback (most recent call last)
<ipython-input-55-e1ee1fab41d5> in <module>()
1 # Create model object in inference mode.
2 get_ipython().magic('tensorflow_version 1.14')
----> 3 model = modellib.MaskRCNN(mode="inference", model_dir=MODEL_DIR, config=config)
4
5 # Load weights trained on MS-COCO
5 frames
/usr/local/lib/python3.6/dist-packages/tensorflow/python/autograph/impl/api.py in wrapper(*args, **kwargs)
256 except Exception as e: # pylint:disable=broad-except
257 if hasattr(e, 'ag_error_metadata'):
--> 258 raise e.ag_error_metadata.to_exception(e)
259 else:
260 raise
AttributeError: in user code:
/content/Mask_RCNN/mrcnn/model.py:390 call *
roi_level = log2_graph(tf.sqrt(h * w) / (224.0 / tf.sqrt(image_area)))
/content/Mask_RCNN/mrcnn/model.py:341 log2_graph *
return tf.log(x) / tf.log(2.0)
AttributeError: module 'tensorflow' has no attribute 'log'
您能帮我整理一下吗?
答案 0 :(得分:0)
到目前为止,当Matterport版本正在使用Tensorflow 1.X时,您似乎正在使用Tensorflow 2.X版本(即使存在一些与tf2兼容的版本)。 如果要保持这种方式,可以将所有“ tf.log”替换为“ tf.math.log”,也可以将“ tf.random_shuffle”替换为“ tf.random.shuffle”。
您应该检查存储库中的问题,社区非常支持,尤其要检查与您的问题相对应的一项:https://github.com/matterport/Mask_RCNN/issues/1797。 祝你好运。