Mask_RCNN / samples / demo.ipynb问题(AttributeError:模块'tensorflow'没有属性'log')

时间:2020-08-28 14:34:47

标签: python tensorflow

当我尝试在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'

您能帮我整理一下吗?

1 个答案:

答案 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。 祝你好运。