没有名为＆＃x27; tensorflow.keras＆＃x27;的模块

发布于 2025-01-21 20:09:59 字数 754 浏览 0 评论 0原文

我正在尝试使用构建验证模型的自定义对象检测模型进行介绍。我想要的只是我的模型是检测图片中的特定徽标。问题是，我遵循的指南是库存在问题。

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导入检测模型元素时，我的错误即将出现。我在Google Colab上这样做，我检查了版本，它们似乎都是最新的。

ModuleNotFoundError: No module named 'tensorflow.keras'

有什么想法吗？我已经环顾了类似问题，但是我无法解决问题。这似乎不是一个张量的问题，因为我能够构建模型，直到添加ImageAi。

原文

I am trying to play around with a custom object detection model that builds of a pretrained model. All I want is for my model is to detect a specific logo in a picture. The problem is, the guide that I am following is having problems with the libraries.

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()

My error is coming when I import imageai.Detection.Custom import DetectionModelTrainer. I am doing this on google colab and I checked the versions and they seem to be all up to date.

ModuleNotFoundError: No module named 'tensorflow.keras'

Any ideas? I have looked around stack for similar problems yet I haven't been able to resolve my issue. It doesn't seem to be a tensorflow problem since I am able to build my models just fine until I add imageai.

分享到QQ

分享到微博