使用OpenCV+Python構(gòu)建運(yùn)動熱圖視頻
OpenCV是一個強(qiáng)大的圖像和視頻處理庫,在這篇文章中,我將創(chuàng)建一個運(yùn)動熱圖,用于檢測運(yùn)動、以及物體或人的流動方向,在投影公共區(qū)域時可以對建筑師有所幫助。
簡介
OpenCV,或(開源計(jì)算機(jī)視覺)是英特爾于1999年開發(fā)的一個庫,主要提供計(jì)算機(jī)視覺和實(shí)時視頻的相關(guān)操作,它是用C++編寫的,同時也支持多種其它語言(包括Python)。工作流程這個程序是基于一種被稱為高斯背景差法的技術(shù),這項(xiàng)技術(shù)被廣泛應(yīng)用于用穩(wěn)定的攝像機(jī)檢測運(yùn)動物體。背景差法創(chuàng)建一個表示幀(圖像的靜態(tài)部分)背景的模板,對于每一幀,它將減去前一幀。讓我們對該算法的兩個主要步驟進(jìn)行一個簡要概述:背景初始化:在第一步中,通過凍結(jié)第一幀來計(jì)算背景的模型。更新:在第二步中,下一幀將減去上一幀,如果兩幀之間發(fā)生變化(移動),則這些幀的差異將反映出該變化,可以通過應(yīng)用過濾器來進(jìn)行提取差異信息。以下是背景差法應(yīng)用于從城市攝像機(jī)錄制的短視頻的示例:
代碼對于整個項(xiàng)目存儲庫,在此處可以獲取。https://github.com/robertosannazzaro/motion-h(huán)eatmap-opencv/blob/master/README.md代碼首先讀取輸入的視頻文件并初始化所需的一些變量:capture = cv2.VideoCapture('input.mp4')background_subtractor = cv2.bgsegm.createBackgroundSubtractorMOG()length = int(capture.get(cv2.CAP_PROP_FRAME_COUNT))然后,for循環(huán)遍歷視頻幀:for i in range(0, length):
ret, frame = capture.read()
# If first frame if first_iteration_indicator == 1:
first_frame = copy.deepcopy(frame) height, width = frame.shape[:2] accum_image = np.zeros((height, width), np.uint8)
第一個if塊檢查該幀是否為視頻的第一幀,這樣做是為了初始化背景差法的背景,然后accum_image使用與該幀大小相對應(yīng)的大小來初始化該數(shù)組。
filter = background_subtractor.a(chǎn)pply(frame) # remove the background
threshold = 2maxValue = 2ret, th1 = cv2.threshold(filter, threshold, maxValue, cv2.THRESH_BINARY)
accum_image = cv2.a(chǎn)dd(accum_image, th1)
color_image_video = cv2.a(chǎn)pplyColorMap(accum_image, cv2.COLORMAP_HOT)
為了消除例如風(fēng),小鳥飛行等少量運(yùn)動,將閾值與maxValue一起應(yīng)用到遮罩上。然后將掩碼的結(jié)果添加到accum_image數(shù)組中,對每個幀執(zhí)行此操作,結(jié)果由用于存儲視頻中發(fā)生的每個運(yùn)動的accum_image數(shù)組組成。此外,在最后,當(dāng)已經(jīng)對每個幀執(zhí)行了先前描述的操作時,顏色映射被應(yīng)用于遮罩并且遮罩與當(dāng)前幀合并。
更進(jìn)一步,可以制作一個顯示熱圖逐幀衰減的視頻。為了實(shí)現(xiàn)這一點(diǎn),將導(dǎo)出每個幀,然后再次使用cv2,通過合并所有幀來生成視頻:video = cv2.VideoWriter('output.a(chǎn)vi', fourcc, 30.0, (width, height))for image in images: video.write(cv2.imread(os.path.join(image_folder, image)))
cv2.destroyAllWindows()最后結(jié)果

請輸入評論內(nèi)容...
請輸入評論/評論長度6~500個字
最新活動更多
-
3月27日立即報(bào)名>> 【工程師系列】汽車電子技術(shù)在線大會
-
4月30日立即下載>> 【村田汽車】汽車E/E架構(gòu)革新中,新智能座艙挑戰(zhàn)的解決方案
-
5月15-17日立即預(yù)約>> 【線下巡回】2025年STM32峰會
-
即日-5.15立即報(bào)名>>> 【在線會議】安森美Hyperlux™ ID系列引領(lǐng)iToF技術(shù)革新
-
5月15日立即下載>> 【白皮書】精確和高效地表征3000V/20A功率器件應(yīng)用指南
-
5月16日立即參評 >> 【評選啟動】維科杯·OFweek 2025(第十屆)人工智能行業(yè)年度評選
推薦專題
- 1 UALink規(guī)范發(fā)布:挑戰(zhàn)英偉達(dá)AI統(tǒng)治的開始
- 2 北電數(shù)智主辦酒仙橋論壇,探索AI產(chǎn)業(yè)發(fā)展新路徑
- 3 降薪、加班、裁員三重暴擊,“AI四小龍”已折戟兩家
- 4 “AI寒武紀(jì)”爆發(fā)至今,五類新物種登上歷史舞臺
- 5 國產(chǎn)智駕迎戰(zhàn)特斯拉FSD,AI含量差幾何?
- 6 光計(jì)算迎來商業(yè)化突破,但落地仍需時間
- 7 東陽光:2024年扭虧、一季度凈利大增,液冷疊加具身智能打開成長空間
- 8 地平線自動駕駛方案解讀
- 9 封殺AI“照騙”,“淘寶們”終于不忍了?
- 10 優(yōu)必選:營收大增主靠小件,虧損繼續(xù)又逢關(guān)稅,能否乘機(jī)器人東風(fēng)翻身?