使用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()
以下是使用上述代碼進行對象檢測的一些結果。
蘋果和書
香蕉和蘋果
多種水果