python 用ImageAI绘制边界框

h7wcgrx3  于 2023-03-16  发布在  Python
关注(0)|答案(1)|浏览(126)

我正在使用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)。
你能建议我修复这个错误吗?谢谢。

oxcyiej7

oxcyiej71#

您必须下载它并在model_path中设置文件的路径。例如,如果文件与脚本位于同一文件夹中,则路径为'./resnet50_coco_best_v2.0.1.h5'

相关问题