1. 程式人生 > 程式設計 >10行Python程式碼計算汽車數量的實現方法

10行Python程式碼計算汽車數量的實現方法

當你還是個孩子坐車旅行的時候,你玩過數經過的汽車的數目的遊戲嗎?

在這篇文章中,我將教你如何使用10行Python程式碼構建自己的汽車計數程式。以下是環境及相應的版本庫:

  • Python版本 3.6.9
  • cvlib: 0.2.2
  • opencv-python: 4.1.1.26
  • tensorflow: 1.14.0
  • matplotlib: 3.1.1
  • Keras: 2.2.5

下面的程式碼用於匯入所需的python庫、從儲存中讀取影象、對影象執行目標檢測、用邊界框顯示影象以及關於檢測目標的標籤、計算影象中的汽車數量並列印影象:

import cv2
import matplotlib.pyplot as plt
import cvlib as cv
from cvlib.object_detection import draw_bbox
im = cv2.imread('cars_4.jpeg')
bbox,label,conf = cv.detect_common_objects(im)
output_image = draw_bbox(im,bbox,conf)
plt.imshow(output_image)
plt.show()
print('Number of cars in the image is ' str(label.count('car')))

此影象的輸出:

Number of cars in the image is 29

此影象的輸出:

Number of cars in the image is 22

此影象的輸出:

Number of cars in the image is 25

你的汽車計數器程式現在已經準備好了。你可以用它來做一些有趣的實驗,比如計算每天經過你家門前車道的汽車數量。

要了解更多關於cvlib庫的資訊,可以訪問下面的連結。

cvlib: https://www.cvlib.net

歡迎關注PyTorch官方中文教程站:http://pytorch.panchuang.net/

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。