close

點擊下方卡片,關注「新機器視覺」公眾號

視覺/圖像重磅乾貨,第一時間送達


當今時代最令人頭疼的事情就是找不到停車位,尤其是找20分鐘還沒有找到停車位。

根據複雜性和效率的不同,任何問題都具有一個或多個解決方案。目前智能停車系統的解決方案,主要包括基於深度學習實現,以及基於重量傳感器、光傳感器實現等。

本期我們將一起通過使用攝像頭和少量代碼來實現最簡單的智能停車系統。該解決方案所使用的概念非常簡單。它由具有以下兩個腳本組成:

1.選擇停車位的坐標並將其保存到文件中。
2.從文件中獲取坐標,並確定該點是否可用。

將該解決方案分成兩個腳本的原因是,避免在每次確定是否有可用停車位的時候,就進行停車位的選擇。

為了使這一過程儘可能簡單,從現在開始,我們將這兩個腳本稱為selector和detector。

相關依賴

在本文中,我們使用python 3.7.6,但其他版本(例如3.6或3.8)當然也可以使用。首先我們要檢查python的版本,我們通過在控制台中編寫python –version,即可返回已安裝的python版本。

C:\Users\Razvan>python --versionPython 3.7.6

在開始構建該系統依賴項之前,我們可以設置一個虛擬環境。通過以下鏈接我們可以了解更多有關虛擬環境的信息https://docs.python.org/3.7/tutorial/venv.html。

也可以使用conda創建和管理環境。有關更多信息見https://docs.anaconda.com/anaconda/。

在python中設置完所有內容後, 最重要的依賴關係將是OpenCV庫。通過pip將其添加到虛擬環境中,可以運行pip install opencv-python。

要檢查所有設置是否正確,我們可以使用以下cv2.__version__命令打印環境中可用的當前OpenCV版本。

(OpenCV) C:\Users\Razvan>pythonPython 3.7.6 (default, Jan 8 2020, 20:23:39) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32Type "help", "copyright", "credits" or "license" for more information.>>> import cv2>>> print(cv2.__version__)4.2.0>>>

在第一行中,我們可以看到在該項目中使用了名為OpenCV的虛擬環境。

步驟

首先,我們需要安裝一個停車場攝像頭。由於我們並沒有一個窗戶可以看到的任何停車場,因此我們選擇使用舊汽車玩具和印刷紙。另外,我在停車場上方設置了一個網絡攝像頭,以獲取良好的圖像,因此我們正在處理的圖像如下所示:

selector選擇器

接下來,我們來介紹編碼部分。首先,我們需要構建選擇器。我們從導入所需模塊開始

import cv2import csv

之後,我們開始獲取圖像,在該圖像上選擇停車位。為此,我們可以選擇攝網絡攝像頭提供的第一幀,保存並使用該圖像選擇停車位。下面的代碼是這樣的:

1.打開image變量中的視頻流;suc確定流是否成功打開。
2.將第一幀寫入frame0.jpg。
3.流被釋放,所有窗口都關閉。
4.新保存的圖片將以img變量形式讀取。
VIDEO_SOURCE = 1cap = cv2.VideoCapture(VIDEO_SOURCE)suc, image = cap.read()cv2.imwrite("frame0.jpg", image)cap.release()cv2.destroyAllWindows()img = cv2.imread("frame0.jpg")

現在,我們已經保存了第一幀並在img變量中將其打開,可以使用selectROIs函數標記停車位。ROI被定義為感興趣的區域,代表圖像的一部分,我們將在其上應用不同的函數以及濾波器來獲取結果。

返回到selectROIs函數,這將返回一個列表(類型:numpy.ndarray),其中包含我們組裝圖像所需的數字及其邊界ROI。

r = cv2.selectROIs('Selector', img, showCrosshair = False, fromCenter = False)

我們的列表將保存在變量r中。賦予cv2.selectROIs函數的參數如下:

1.「選擇器」是允許我們選擇投資回報率的窗口的名稱。
2.img是包含我們要選擇圖像的變量。
3.showCrosshair = Flase刪除選區內部的中心線。可以將其設置為True,因為對結果沒有影響。
4.fromCenter = False是一個非常重要的參數,因為如果將其設置為True,則正確的選擇會困難得多。

選擇所有停車位之後,是時候將它們寫入.csv文件了。為此,我們需要將r變量轉換為python列表,可以使用rlist = r.tolist()命令實現。

擁有適當的數據後,我們將其保存到.csv文件中,以備將來使用。

with open('data/rois.csv', 'w', newline='') as outf: csvw = csv.writer(outf) csvw.writerows(rlist)
detector探測器

現在我們已經選擇了停車位,是時候進行一些圖像處理了。解決這個問題的方法如下:

1.從.csv文件獲取坐標。
2.從中構建新圖像。
3.應用OpenCV中可用的Canny函數。
4.計算新圖像內的白色像素。
5.建立一個點內的像素範圍將被占用。
6.在實時供稿上繪製一個紅色或綠色矩形。

對於所有這些操作,我們需要定義一個要應用於每個位置的函數。該函數如下所示:

def drawRectangle(img, a, b, c, d): sub_img = img[b:b + d, a:a + c] edges = cv2.Canny(sub_img, lowThreshold, highThreshold) pix = cv2.countNonZero(edges) if pix in range(min, max): cv2.rectangle(img, (a, b), (a + c, b + d), (0, 255, 0), 3) spots.loc += 1 else: cv2.rectangle(img, (a, b), (a + c, b + d), (0, 0, 255), 3)

現在我們已經實現了所需的功能,如果我們直接將其應用於.csv文件中的每組坐標效果可能並不好。因此我們做如下處理

首先,我們的有一些參數如果可以在腳本運行時(也可以在通過GUI)實時調整它們,那就更好了。為此,我們需要構建一些軌跡欄。OpenCV為我們提供這項功能。

我們需要一個回調函數,該函數不執行任何操作,但作為使用OpenCV創建軌跡欄的參數是必需的。實際上,回調參數具有明確定義的用途,但我們在此不使用它。要了解有關此內容的更多信息,查閱OpenCV文檔。

def callback(foo): pass

現在我們需要創建軌跡欄。我們將使用cv2.namedWindow和cv2.createTrackbar功能。

cv2.namedWindow('parameters')cv2.createTrackbar('Threshold1', 'parameters', 186, 700, callback)cv2.createTrackbar('Threshold2', 'parameters', 122, 700, callback)cv2.createTrackbar('Min pixels', 'parameters', 100, 1500, callback)cv2.createTrackbar('Max pixels', 'parameters', 323, 1500, callback)

現在,我們已經創建了用於操作參數的GUI,只剩下一件事了。這就是圖像中可用斑點的數量。在drawRectangle中定義為spot.loc。這是一個靜態變量,必須在程序開始時進行定義。該變量為靜態變量的原因是,我們希望調用的每個drawRectangle函數都將其寫入相同的全局變量,而不是每個函數都使用一個單獨的變量。這樣可以防止返回的可用空間數量大於實際的可用空間數量。

為了實現這一點,我們只需要使用它的loc靜態變量創建spots類。

class spots: loc = 0

現在我們已經準備就緒,只需要從.csv文件中獲取數據,將其所有數據轉換為整數,然後在無限循環中應用構建的函數即可。

with open('data/rois.csv', 'r', newline='') as inf: csvr = csv.reader(inf) rois = list(csvr)rois = [[int(float(j)) for j in i] for i in rois]VIDEO_SOURCE = 1cap = cv2.VideoCapture(VIDEO_SOURCE)while True: spots.loc = 0 ret, frame = cap.read() ret2, frame2 = cap.read() min = cv2.getTrackbarPos('Min pixels', 'parameters') max = cv2.getTrackbarPos('Max pixels', 'parameters') lowThreshold = cv2.getTrackbarPos('Threshold1', 'parameters') highThreshold = cv2.getTrackbarPos('Threshold2', 'parameters') for i in range(len(rois)): drawRectangle(frame, rois[i][0], rois[i][1], rois[i][2], rois[i][3]) font = cv2.FONT_HERSHEY_SIMPLEX cv2.putText(frame, 'Available spots: ' + str(spots.loc), (10, 30), font, 1, (0, 255, 0), 3) cv2.imshow('Detector', frame) canny = cv2.Canny(frame2, lowThreshold, highThreshold) cv2.imshow('canny', canny) if cv2.waitKey(1) & 0xFF == ord('q'): breakcap.release()cv2.destroyAllWindows()
拓展

在我們的循環中實際上只是調用的構建函數要複雜一點。

首先,我們將空間的數量初始化為0,以防止每幀添加數字。

其次,我們進入兩個處理流以顯示真實圖像和已處理的圖像。這有助於更好地了解此腳本的工作方式以及圖像的處理方式。

然後,我們需要在每次迭代中獲取我們創建的參數GUI中的參數值。這是通過cv2.getTrackbarPos功能完成的。

接下來最重要的部分,將drawRectangle函數應用到Selector腳本獲取的所有坐標上。

最後,在結果圖像上寫下可用斑點的數量,顯示Canny函數的結果,顯然,這是一種眾所周知的停止循環的方法。

我們現在便完成了一個智能停車項目!

總結

如今,智能停車已成為熱門話題之一,並且有許多實現方式可以導致良好的功能系統。我們這處理方法並不是完美的,有許多方法可以更好地優化結果,並且可以在更多情況下使用。但是,即使這不能解決停車場危機,也可能是導致危機 的主要原因。


本文僅做學術分享,如有侵權,請聯繫刪文。

—THE END—
arrow
arrow
    全站熱搜
    創作者介紹
    創作者 鑽石舞台 的頭像
    鑽石舞台

    鑽石舞台

    鑽石舞台 發表在 痞客邦 留言(0) 人氣()