10行Python代码搞定目标检测

from imageai.Detection import ObjectDetection #导入了 ImageAI 目标检测类
import os
execution_path = os.getcwd()  #定义了一个变量用来保存我们的 python 文件

detector = ObjectDetection() #定义了目标检测类
detector.setModelTypeAsRetinaNet() #模型的类型设置为 RetinaNet
detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))  #将模型路径设置为 RetinaNet 模型的路径
detector.loadModel() #模型加载到的目标检测类
#调用目标检测函数，解析输入的和输出的图像路径。
detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image.jpg"), output_image_path=os.path.join(execution_path , "imagenew.jpg"))



for eachObject in detections:
    print(eachObject["name"] + " : " + eachObject["percentage_probability"] ) #打印出所检测到的每个目标的名称及其概率值。