imageai - 导入错误 : cannot import name convert_all_kernels_in_model

Question

我正在尝试使用 ImageAI 的模型训练来训练 AI 模型。

这是代码:

 from imageai.Prediction.Custom import ModelTraining
 model_trainer = ModelTraining()
 model_trainer.setModelTypeAsResNet()
 model_trainer.setDataDirectory("idenprof")
 model_trainer.trainModel(num_objects=2, num_experiments=3, enhance_data=True, 
 batch_size=32, show_network_summary=True)

这是我在运行时遇到的错误:

Traceback (most recent call last):
  File ".../FirstTraining.py", line 1, in <module>
    from imageai.Prediction.Custom import ModelTraining
  File ".../lib/python2.7/site-packages/imageai/Prediction/Custom/__init__.py", line 4, in <module>
    from ..DenseNet.densenet import DenseNetImageNet121
  File ".../PycharmProjects/bonez/venv/lib/python2.7/site-packages/imageai/Prediction/DenseNet/densenet.py", line 21, in <module>
    from tensorflow.python.keras.utils import convert_all_kernels_in_model
ImportError: cannot import name convert_all_kernels_in_model

我到处搜索，但找不到相同的问题或解决问题的方法。我安装了以下依赖项:Tensorflow、OpenCV、Keras 和 ImageAI。

Answer 1

更新:原来 ImageAI 还不支持 Tensorflow 2。以下 tensorflow 版本不会发生此问题:pip install tensorflow==1.15.2

---

我遇到了同样的问题，并通过将 ImageAI 库中的所有 tensorflow.python.keras 导入替换为 tensorflow.keras 来解决。在此之后，from imageai.Prediction.Custom import ModelTraining 导入工作正常。

如果您想关注，ImageAI 中有一个 Unresolved 问题:https://github.com/OlafenwaMoses/ImageAI/issues/494