python - Colab 上没有名为“tensorflow.compat.v2”的模块

Question

我正在关注Mask RCNN 的https://thebinarynotes.com/how-to-train-mask-r-cnn-on-the-custom-dataset/教程，并尝试在 Google Colab 上执行它。

Tensorflow和Keras的版本如下： tensorflow==1.13.1 keras==2.1.0

3 周前，我已经使用此代码并成功地在我的自定义数据集上训练了模型，并预测了结果。

但是现在，当我尝试在同一环境中执行相同的代码时，出现以下错误。尝试了多种解决方案，例如将 Tensorflow 更新到 2.x，但代码与它不兼容，并产生了其他问题。在此处输入图像描述

标签： pythontensorflowkeras

Answer 1

该tf.compat.v2模块已添加到1.14. 升级到1.14、1.15或2.0，这将正常工作。

tensorflow.compat.v2在 TensorFlow==1.15 上为我工作

!pip install tensorflow==1.15
import tensorflow.compat.v2 as tf