我正在尝试使用构建预训练模型的自定义对象检测模型。所有我想要的是我的模型是检测一个特定的标志在图片中。问题是，我所遵循的指南在使用库时出现了问题。
import tensorflow as tf
from imageai.Detection.Custom import DetectionModelTrainer
trainer = DetectionModelTrainer()
trainer.setModelTypeAsYOLOv3()
trainer.setDataDirectory(data_directory="/content/drive/MyDrive/Logo_Model2/")
trainer.setTrainConfig(object_names_array=["logo"], batch_size=4, num_experiments=122, train_from_pretrained_model="/content/drive/MyDrive/pretrained-yolov3.h5")
trainer.trainModel()
当我导入imageai.Detection.Custom导入DetectionModelTrainer时，我的错误来了。我在google colab上做这个，我检查了版本，它们似乎都是最新的。
ModuleNotFoundError: No module named 'tensorflow.keras'
任何想法?我已经环顾堆栈类似的问题，但我还没有能够解决我的问题。这似乎不是一个张量流问题，因为我能够建立我的模型，直到我添加imageai。
似乎是最新的tensorflow==2.8.0的一个问题。git问题
现在，你可以恢复到旧版本的tensorflow
pip install tensorflow==2.7
并升级imageAI:
pip install imageai --upgrade