使用ImageAI绘制边界框遇模型路径无效错误的解决咨询

问题场景 #

使用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已存在，但仍报错。

修复方法 #

• 使用绝对路径
  相对路径受脚本运行目录影响，直接指定模型文件的绝对路径能避免路径解析错误。例如：

  # Windows示例
  model_path = 'D:/ai_projects/models/resnet50_coco_best_v2.0.1.h5'
  # Linux/macOS示例
  model_path = '/home/user/ai_models/resnet50_coco_best_v2.0.1.h5'
  

• 确认脚本工作目录
  在代码开头添加以下代码，查看脚本实际运行的目录，判断模型文件是否处于该目录下：

  import os
  print("当前工作目录:", os.getcwd())
  

  如果模型不在此目录，要么将模型移动到该目录，要么调整model_path为相对该目录的正确路径（比如模型在子目录models下，就写model_path = 'models/resnet50_coco_best_v2.0.1.h5'）。

• 检查文件名准确性
  确保文件名完全匹配，包括大小写（Linux/macOS系统严格区分文件名大小写），避免出现拼写错误或大小写不一致的情况。

内容的提问来源于stack exchange，提问作者Alena