ImageAI (二) 使用Python快速简单实现物体检测 Object Detection

冷不防 2021-08-19 23:17 429阅读 0赞

上一篇已经讲解了ImageAI实现图片预测的方法，现在再来讲解一下ImageAI的第二个功能物体检测。  
ImageAI提供了非常方便和强大的方法来对图像执行对象检测并从图像中提取每个对象。 ImageAI使用的模型是RetinaNet，并提供了已经训练完成的模型文件。  
同样，仅需几行代码就能完成物体检测的过程。  
ImageAI [github地址][github]  
准备工作以及ImageAI的安装可以详见上一篇 [ImageAI (一)][ImageAI]  
使用版本 ImageAI-2.1.0

--------------------

**Object Detection**  
训练好的RetinaNet模型：[RetinaNet][]

**detection.py**

from imageai.Detection import ObjectDetection
    import os
    import time
    #计时
    start = time.time()
    
    
    execution_path = os.getcwd()
    
    detector = ObjectDetection()
    detector.setModelTypeAsRetinaNet()
    
    #载入已训练好的文件
    detector.setModelPath( os.path.join(execution_path , "resnet50_coco_best_v2.0.1.h5"))
    detector.loadModel()
    
    #将检测后的结果保存为新图片
    detections = detector.detectObjectsFromImage(input_image=os.path.join(execution_path , "image3.jpg"), output_image_path=os.path.join(execution_path , "image3new.jpg"))
    
    #结束计时
    end = time.time()
    
    for eachObject in detections:
        print(eachObject["name"] ," : " ,eachObject["percentage_probability"] , " : ", eachObject["box_points"] )  ##预测物体名:预测概率:物体两点坐标（左上，右下）
        print("--------------------------------")
    
    print ("\ncost time:",end-start)

下面是github提供的图片  
![这里写图片描述][70]

我跑出来的结果  
![这里写图片描述][70 1]

person  :  76.33113265037537  :  [463 139 512 221]
    --------------------------------
    person  :  83.83048176765442  :  [600 131 639 213]
    --------------------------------
    person  :  96.08134031295776  :  [534 102 579 225]
    --------------------------------
    person  :  96.66982889175415  :  [  8 103  63 248]
    --------------------------------
    motorcycle  :  71.07154726982117  :  [273 180 346 306]
    --------------------------------
    dog  :  94.58073377609253  :  [398 314 444 433]
    --------------------------------
    car  :  55.41401505470276  :  [215 140 388 299]
    --------------------------------
    person  :  86.19718551635742  :  [154 145 255 393]
    --------------------------------
    
    cost time: 13.959604263305664

我自己随便找的一张图  
![这里写图片描述][70 2]  
结果  
![这里写图片描述][70 3]

person  :  61.145514249801636  :  [175 580 202 650]
    --------------------------------
    person  :  63.822001218795776  :  [210 591 237 680]
    --------------------------------
    person  :  93.19865703582764  :  [314 592 345 688]
    --------------------------------
    person  :  71.38553857803345  :  [ 54 585 103 682]
    --------------------------------
    person  :  80.48739433288574  :  [272 594 309 692]
    --------------------------------
    motorcycle  :  71.90784811973572  :  [ 36 626  98 704]
    --------------------------------
    handbag  :  55.64073324203491  :  [818 654 876 764]
    --------------------------------
    person  :  92.45884418487549  :  [650 575 774 793]
    --------------------------------
    person  :  83.37823152542114  :  [787 597 880 796]
    --------------------------------
    person  :  96.3289201259613  :  [1096  586 1204  790]
    --------------------------------
    
    cost time: 15.170496702194214

完成！

[github]: https://github.com/OlafenwaMoses/ImageAI/tree/master/imageai/Detection
[ImageAI]: https://blog.csdn.net/zkt286468541/article/details/81007022
[RetinaNet]: https://github.com/OlafenwaMoses/ImageAI/releases/download/1.0/resnet50_coco_best_v2.0.1.h5
[70]: /images/20210726/7ad0885ac79c4f80a0abcfdc3370998d.png
[70 1]: /images/20210726/f789bc9091a6496cba79c6c7c5f22a04.png
[70 2]: /images/20210726/97fc8ece7a0443e88088a05246194abe.png
[70 3]: /images/20210726/f5e5958ca1ea472c87a50b0dd22b7ba9.png