家里烤箱忘關(guān)炕吸?用Python和OpenCV寫個程序來幫忙提醒蒸矛!

這篇文章應(yīng)用計算機視覺和圖像處理技術(shù)劣像,展示了檢測烤箱開關(guān)狀態(tài)的過程屋群。在生活中对竣,有時你會粗心大意忘關(guān)烤箱之類的廚房電器烦味,這潛在很大的危險谬俄。因此作者采用 Python 和 OpenCV,通過家庭攝像頭獲取的圖像來自動識別烤箱是否開著弃理,進而可以觸發(fā)警報溃论。

“我忘關(guān)烤箱了嗎?”

Python學習群:593088321

這個問題常常會在最不方便的時候出現(xiàn)在你的腦子里痘昌。

有時是當你剛剛走出家門的時候钥勋。 有時是當你在上班路上的時候。 有時是當你坐在飛機上準備度個長假的時候……

解決這個問題的方法是多種多樣的:

橡皮筋的方法

大聲說或唱出來(就像 Samuel L. Jackson 一樣)

當你離開家去度假的時候辆苔,給烤箱在內(nèi)的電器列個清單或做個標記算灸。 或者,我們也許有更好的做法……

在本教程中驻啤,我們會嘗試利用技術(shù)手段解決這個問題菲驴。

Github 里有完整代碼。

問題定義

對我們而言骑冗,我們需要確定一個信號赊瞬,用于判斷烤箱的開關(guān)狀態(tài)先煎。在我的廚房里,這個信號就是頂部標記著“烤箱開”字樣的紅色燈巧涧。

當紅色燈亮的時候薯蝎,烤箱是開著的:

Python學習群:593088321

當紅色燈滅的時候,烤箱是關(guān)著的:

預(yù)備條件

確保你的電腦上安裝了以下應(yīng)用:

OpenCV 3.0

Python 2.7

Numpy 1.9

安裝 OpenCV3.0 和 Python 2.7

如果你尚未安裝 OpenCV谤绳,請按照 Adrian Rosebrock 的完美教程占锯,在 OSX 系統(tǒng)上安裝 OpenCV 3.0 和 Python 2.7+。我在安裝步驟里增加了一些自己的注釋闷供,以防你在 OSX 上編譯 OpenCV 3.0時遇到問題烟央。

步驟

如果你已經(jīng)成功在你的環(huán)境中安裝了 OpenCV,我們就可以開始判斷烤箱開關(guān)的數(shù)據(jù)分析了歪脏。

加載需要的包

argparse —— 參數(shù)處理庫疑俭。

numpy —— 高度優(yōu)化的數(shù)值運算庫。OpenCV 在數(shù)組結(jié)構(gòu)中使用 numpy婿失。

cv2 —— OpenCV 中圖像處理庫钞艇。

import?argparse

import?numpy?as?np

import?cv2

載入圖片

image = cv2.imread(image_path)

圖片降噪

我們?yōu)榱私o圖片降噪,需要對輸入圖片進行平滑處理豪硅。這樣會更容易在圖片中定位目標哩照。使用 medianBlur 函數(shù),把光圈大小定為 3 懒浮。數(shù)字越大意味著圖像會越模糊飘弧。

blur_image = cv2.medianBlur(image, 3)

把圖片顏色轉(zhuǎn)為 HSV 格式

HSV —— 色度、飽和度和純度(亮度)砚著。HSV 可以讓我們提取出一個彩色對象次伶,因為它比 BGR 格式(譯者注:與我們常說的RBG色彩模型類似,三個字母分別代表紅藍綠三色)更容易表征顏色稽穆。

把圖片轉(zhuǎn)為 HSV 格式可以讓我們通過色度(一個值而不是三個值)冠王,來確定圖片中的一個顏色。

以下是實現(xiàn)方式:

hsv_image = cv2.cvtColor(blur_image, cv2.COLOR_BGR2HSV)

執(zhí)行結(jié)果如下圖:

Python學習群:593088321

檢測圖片中的顏色

為了檢測我們想要的顏色舌镶,我們可以查看烤箱燈圖片中顏色的直方圖柱彻。

Python學習群:593088321
Python學習群:593088321

我們可以看出餐胀,紅色在圖片中占統(tǒng)治地位。紅色有兩個高峰 —— 一個幅度高卖擅,一個幅度低。這些顏色值轉(zhuǎn)變?yōu)樯确秶鷱?0 到10,以及從 160 到 180 (針對紅色)。

在 HSV 圖片中針對每一個色度范圍琳猫,我們可以創(chuàng)建一個遮罩,來去掉所有不在選定范圍的無關(guān)顏色私痹。

def?create_hue_mask(image, lower_color, upper_color):

lower = np.array(lower_color, np.uint8)

upper = np.array(upper_color, np.uint8)

# Create a mask from the colors

mask = cv2.inRange(image, lower, upper)

output_image = cv2.bitwise_and(image, image, mask = mask)

return?output_image

# Get lower red hue

lower_red_hue = create_hue_mask(hsv_image, [0, 100, 100], [10, 255, 255])

# Get higher red hue

higher_red_hue = create_hue_mask(hsv_image, [160, 100, 100], [179, 255, 255])

結(jié)果如下:

接下來把這些圖片合并在一起紊遵,以抓取所有紅色色度暗膜。

full_image = cv2.addWeighted(lower_red_hue, 1.0, higher_red_hue, 1.0, 0.0)

結(jié)果如下:

Python學習群:593088321

發(fā)現(xiàn)圖片中的圓圈

現(xiàn)在我們的圖片上僅有紅色色度学搜,接著我們需要判定紅燈是否開啟(即是否存在紅色色度的那個圓圈)瑞佩。我們需要在新圖中發(fā)現(xiàn)圓圈炬丸,不過首先需要把圖片轉(zhuǎn)換成灰度圖(因為 HoughCircles 函數(shù)的輸入要求是灰度圖)。

檢測圖片中的圓圈需要以下參數(shù)(使用 OpenCV中 的 HoughCircles 函數(shù)):

灰度圖輸入焕阿。

HOUGH_GRADIENT 是用來檢測圓圈的方法(目前僅有的一個方法)捣鲸。

累加器和圖片分辨率的反比栽惶。在本例中外厂,為1.2代承。

待檢測圓圈圓心的最小距離,本例中為100墓律。

#Convert image to grayscale

image_gray = cv2.cvtColor(full_image, cv2.COLOR_BGR2GRAY)

#Find circles in the image

circles = cv2.HoughCircles(image_gray, cv2.HOUGH_GRADIENT, 1.2, 100)

結(jié)果

此時就可以檢查是否有圓圈了耻讽。如果有就意味著至少有一個烤箱燈亮著帕棉。如果找不到圓圈就意味著沒有燈亮香伴,烤箱關(guān)著具帮。

為了證明此結(jié)論,我們可以用下述代碼在原圖中畫圓圈:

# Draw the circles on the original image

circles = np.round(circles[0, :]).astype("int")

for?(center_x, center_y, radius)?in?circles:

cv2.circle(image, (center_x, center_y), radius, (0, 255, 0), 4)

結(jié)果如下:

Python學習群:593088321

下一步

接下來還有很多可以做的巴比,比如:

檢測特定燈的開啟轻绞,用以了解烤箱的真實狀態(tài)政勃。

建立一個服務(wù)以便遠程檢查烤箱狀態(tài)奸远。

把該功能加入樹莓派(譯者注:基于Linux的迷你開發(fā)板)丸冕,我們就擁有可以警告烤箱關(guān)閉與否的小型設(shè)備胖烛。

全部樣例代碼可以在學習群內(nèi)獲扰宸趟畏!

Python學習群:593088321

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末拱镐,一起剝皮案震驚了整個濱河市沃琅,隨后出現(xiàn)的幾起案子蜘欲,更是在濱河造成了極大的恐慌姥份,老刑警劉巖澈歉,帶你破解...
    沈念sama閱讀 218,204評論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件埃难,死亡現(xiàn)場離奇詭異涡尘,居然都是意外死亡,警方通過查閱死者的電腦和手機细疚,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,091評論 3 395
  • 文/潘曉璐 我一進店門疯兼,熙熙樓的掌柜王于貴愁眉苦臉地迎上來镇防,“玉大人来氧,你說我怎么就攤上這事啦扬。” “怎么了胃榕?”我有些...
    開封第一講書人閱讀 164,548評論 0 354
  • 文/不壞的土叔 我叫張陵,是天一觀的道長换帜。 經(jīng)常有香客問我惯驼,道長隙畜,這世上最難降的妖魔是什么议惰? 我笑而不...
    開封第一講書人閱讀 58,657評論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘彤钟。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 67,689評論 6 392
  • 文/花漫 我一把揭開白布赂弓。 她就那樣靜靜地躺著,像睡著了一般褐奴。 火紅的嫁衣襯著肌膚如雪脖旱。 梳的紋絲不亂的頭發(fā)上币旧,一...
    開封第一講書人閱讀 51,554評論 1 305
  • 那天鳍刷,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 40,302評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,216評論 0 276
  • 序言:老撾萬榮一對情侶失蹤奇徒,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后网持,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體辟汰,經(jīng)...
    沈念sama閱讀 45,661評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,851評論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片绣版。...
    茶點故事閱讀 39,977評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡铸磅,死狀恐怖弧械,靈堂內(nèi)的尸體忽然破棺而出界轩,到底是詐尸還是另有隱情热鞍,我是刑警寧澤,帶...
    沈念sama閱讀 35,697評論 5 347
  • 正文 年R本政府宣布逐工,位于F島的核電站髓涯,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 41,306評論 3 330
  • 文/蒙蒙 一卵皂、第九天 我趴在偏房一處隱蔽的房頂上張望钠惩。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,898評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春拷呆,著一層夾襖步出監(jiān)牢的瞬間腰懂,已是汗流浹背娄蔼。 一陣腳步聲響...
    開封第一講書人閱讀 33,019評論 1 270
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 48,138評論 3 370
  • 正文 我出身青樓目溉,卻偏偏與公主長得像缭付,于是被迫代替她去往敵國和親绣檬。 傳聞我的和親對象是個殘疾皇子娇未,可洞房花燭夜當晚...
    茶點故事閱讀 44,927評論 2 355

推薦閱讀更多精彩內(nèi)容