Create test_detector-01.py

eastmountyxz · web-flow · commit 5d57c7f503e2 · 2021-08-03T18:18:37.000+08:00
diff --git a/blog44-imageAI/Object-detection/test_detector-01.py b/blog44-imageAI/Object-detection/test_detector-01.py
@@ -0,0 +1,34 @@
+# -*- coding: utf-8 -*-
+"""
+Created on Mon Jul 26 14:26:27 2021
+@author: xiuzhang
+"""
+
+from imageai.Detection import ObjectDetection
+
+#实例化
+detector = ObjectDetection()
+
+#路径定义
+model_path = "./models/yolo-tiny.h5"
+input_path = "./input/person-02.png"
+output_path = "./output/person-02.png"
+
+#预训练模式
+detector.setModelTypeAsTinyYOLOv3()
+
+#设置预训练模型路径
+detector.setModelPath(model_path)
+
+#加载模型
+detector.loadModel()
+
+#创建对象
+detection = detector.detectObjectsFromImage(input_image=input_path, 
+                                            output_image_path=output_path)
+
+#检测结果
+for eachItem in detection:
+    print(eachItem["name"] , " : ", eachItem["percentage_probability"])
+    
+    
