我试图训练 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.