imageai3.0.3 detectCustomObjectsFromImage方法使用
时间: 2024-03-24 14:41:30 浏览: 95
`detectCustomObjectsFromImage` 方法是 ImageAI 中用于在图像中检测自定义对象的方法。下面是使用该方法的步骤:
1. 导入必要的库
```python
from imageai.Detection import ObjectDetection
```
2. 创建 ObjectDetection 类的实例,并设置模型路径和自定义对象的类别
```python
detector = ObjectDetection()
model_path = "path/to/model"
custom_objects = detector.CustomObjects(person=True, bicycle=True, motorcycle=True)
detector.setModelTypeAsYOLOv3()
detector.setModelPath(model_path)
detector.loadModel()
```
在上述代码中,我们假设模型是 YOLOv3,并且我们想要检测的自定义对象是人、自行车和摩托车。
3. 调用 `detectCustomObjectsFromImage` 方法,传递图像路径和输出路径(可选)
```python
input_image = "path/to/image"
output_image = "path/to/output/image"
detections = detector.detectCustomObjectsFromImage(input_image, output_image=output_image, custom_objects=custom_objects, minimum_percentage_probability=30)
```
在上述代码中,我们传递了输入图像的路径和输出图像的路径(如果需要),以及我们之前定义的自定义对象和最小百分比概率。
4. 处理输出结果
```python
for detection in detections:
print(detection["name"], " : ", detection["percentage_probability"], " : ", detection["box_points"])
```
在上述代码中,我们遍历检测结果列表,并打印每个检测到的对象的名称、百分比概率和边界框坐标。
以上就是使用 `detectCustomObjectsFromImage` 方法的步骤。需要注意的是,该方法只能检测自定义对象,如果需要检测通用对象,需要使用 `detectObjectsFromImage` 方法。
阅读全文