0

我试图训练 imageai 模型,但是当我尝试加载它以检查我得到值错误错误时,就像我的模型是空的。也许我需要以其他方式保存它?

我在这里训练它:

exec_path = os.getcwd()

model_trainer = ClassificationModelTrainer()
model_trainer.setModelTypeAsResNet()
model_trainer.setDataDirectory("blood-cells", models_subdirectory="models")
model_trainer.trainModel(num_objects=2, num_experiments=10, batch_size=8, enhance_data=True, show_network_summary=True)

在这里,我正在尝试加载它:

exec_path = os.getcwd()

detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
detector.setModelPath(os.path.join(
    exec_path, "blood-cells/models/model_ex-003_acc-1.000000.h5")
)
detector.loadModel()

list = detector.detectObjectsFromImage(
    input_image=os.path.join(exec_path, "blood-cells/test/erythrocytes/image-10.png"),
    output_image_path=os.path.join(exec_path, "new_objects.jpg"),
    minimum_percentage_probability=90,
    display_percentage_probability=True,
    display_object_name=True
)

我得到的错误:

Traceback (most recent call last):
  File "/Users/ilaseregin/Desktop/ImageAI/main.py", line 21, in <module>
    detector.loadModel()
  File "/Users/ilaseregin/Desktop/ImageAI/venv/lib/python3.8/site-packages/imageai/Detection/__init__.py", line 181, in loadModel
    model = retinanet_models.load_model(self.modelPath, backbone_name='resnet50')
  File "/Users/ilaseregin/Desktop/ImageAI/venv/lib/python3.8/site-packages/imageai/Detection/keras_retinanet/models/__init__.py", line 87, in load_model
    return keras.models.load_model(filepath, custom_objects=backbone(backbone_name).custom_objects)
  File "/Users/ilaseregin/Desktop/ImageAI/venv/lib/python3.8/site-packages/tensorflow/python/keras/saving/save.py", line 206, in load_model
    return hdf5_format.load_model_from_hdf5(filepath, custom_objects,
  File "/Users/ilaseregin/Desktop/ImageAI/venv/lib/python3.8/site-packages/tensorflow/python/keras/saving/hdf5_format.py", line 181, in load_model_from_hdf5
    raise ValueError('No model found in config file.')
ValueError: No model found in config file.
4

0 回答 0