Python 10行代码搞定图片中物体检测

使用Python， 如何编程实现图片中的物体检测【也叫对象检测/object detection】？

怎么知道图像中有哪些物体？

或许您想要计算图像中苹果的数量？

在这篇文章中，我将向您展示如何使用Python在不到10行代码中创建自己的对象/物体检测程序。

环境准备，需要安装以下python库：

opencv-python
cvlib

下面是导入所需python库，从存储中读取图像，对图像执行对象检测以及使用边界框显示图像以及检测到的对象标签的代码。

import cv2
import matplotlib.pyplot as plt
import cvlib as cv
from cvlib.object_detection import draw_bbox
im = cv2.imread('apple-256261_640.jpg')
bbox, label, conf = cv.detect_common_objects(im)
output_image = draw_bbox(im, bbox, label, conf)
plt.imshow(output_image)
plt.show()

以下是使用上述代码进行对象检测的一些结果。

苹果和书

香蕉和苹果

多种水果