我正在使用ImageAI在图像上绘制边界框。
from imageai.Detection import ObjectDetection
detector = ObjectDetection()
detector.setModelTypeAsRetinaNet()
# Load model
model_path = 'resnet50_coco_best_v2.0.1.h5'
detector.setModelPath(model_path)
detector.loadModel()
这将引发错误:
----> 6 detector.setModelPath(model_path)
ValueError: invalid path, path not pointing to a valid file.
但是,存在“resnet50_coco_best_v2.0.1.h5”文件(https://github.com/OlafenwaMoses/ImageAI/releases/tag/1.0)。
你能建议我修复这个错误吗?谢谢。
1条答案
按热度按时间oxcyiej71#
您必须下载它并在
model_path
中设置文件的路径。例如,如果文件与脚本位于同一文件夹中,则路径为'./resnet50_coco_best_v2.0.1.h5'