99视频精品,菲律宾毛片,亚洲欧美另类在线,国产五月丁香,美女被桶秘免费网站,国产精品嫩草AV城中村,国产视频aaa,国产swag在线观看

如何使用Mask-RCNN檢測停車位可用性?

空/占用停車位

我最近做了一個項目，根據(jù)安全攝像頭的照片來檢測停車位是否可用或被占用。我的工作有局限性，我將進(jìn)一步詳細(xì)介紹這些局限性，但一旦這些問題得到解決，這個項目可能是一個低成本的解決方案，以優(yōu)化停車位的可用性。安裝這些額外的設(shè)備肯定是有潛力的。

讓我詳細(xì)說明我在這個項目中使用的資源。Kaggle上有一個停車場數(shù)據(jù)集，無論位置是否被占用，它都有足夠的數(shù)據(jù)點來訓(xùn)練一個深度學(xué)習(xí)模型和xml文件，我們可以在這里訪問數(shù)據(jù)集停車場數(shù)據(jù)集。對于模型，我使用了最先進(jìn)的目標(biāo)檢測和分割Mask-RCNN模型，它的性能驚人，可以通過這個鏈接訪問。

如我們所見，Mask-RCNN在COCO數(shù)據(jù)集模型上進(jìn)行了直接即用的預(yù)訓(xùn)練，在目標(biāo)檢測和分割方面做得很好。盡管在某些情況下，它把汽車誤分為火車和卡車。

Mask-RCNN對象檢測和分割

我也嘗試了YOLO-v3，它的性能是一樣的，所以我沒有進(jìn)一步使用YOLO模型，但如果小伙伴正在尋找替代的模型，YOLO是一個很棒的對象檢測模型，這里有鏈接。

YOLO模型檢測停車場上的車輛

首先我使用Mask-RCNN模型來檢測車位上的車輛，并給定可用車位的數(shù)量來計算空車位。我們的模型不需要所有的COCO類，所以我將類限制為汽車、卡車和摩托車。但是預(yù)先訓(xùn)練的COCO數(shù)據(jù)集模型在檢測小物體方面做得并不好，即使我嘗試調(diào)整閾值和邊界框，那些被誤分類為火車的汽車也沒有被檢測到。這里繪制的是邊界框，而不是可視化方法提供的模型。

Mask-RCNN模型預(yù)測

基于數(shù)據(jù)集的預(yù)測，這樣我們就有兩個類來預(yù)測這個位置是被占用了還是空了。這個停車場數(shù)據(jù)集的不足之處是，只從兩個角度拍攝照片，這導(dǎo)致訓(xùn)練模型時存在過擬合問題，沒有很好的泛化能力。其次，每個車位的標(biāo)注不完整，在訓(xùn)練過程中會產(chǎn)生異常，并且并不是所有照片上可用的車位都被標(biāo)注了，這也導(dǎo)致了模型的性能較差。我將向小伙伴展示我的意思，下面的照片只是顯示了有多少停車位被注釋:

并非所有停車位都已標(biāo)注

許多xml文件沒有填充類:

注釋文件中缺少信息

缺少信息的停車位數(shù)量可能會增加到10-15個停車位，但在運(yùn)行一些代碼后，我認(rèn)為這可以手動修復(fù):

有太多的文件需要修復(fù)，只為了訓(xùn)練模型的頂層，避免這些文件比較容易。為了解析xml文件，我使用了內(nèi)置的python包XML .etree.Elementtree。同樣值得注意的是，邊界框的坐標(biāo)是以一個中心點的角度給出的，所以如果我們想要正確地解析和創(chuàng)建邊界框，就必須進(jìn)行一些調(diào)整。

函數(shù)解析xml文件，提取停車位的輪廓

由于Mask-RCNN使用掩碼來訓(xùn)練類，采用與KangarooDetection文章類似的方式，可以在這里訪問，我使用邊界框來創(chuàng)建掩碼。這篇文章實際上幫助我們理解了如何使用Mask-RCNN模型，并且機(jī)器學(xué)習(xí)掌握對于許多機(jī)器學(xué)習(xí)應(yīng)用程序來說是一個很好的資源。如果小伙伴還沒看過的話，可以瀏覽一下。主要的區(qū)別是，我們將檢測兩個類，而不是一個，從而創(chuàng)建基于占用分類的掩碼，以及如何計算邊界框。

函數(shù)在邊界框之外創(chuàng)建掩碼，并創(chuàng)建兩個要檢測的類

至于數(shù)據(jù)集結(jié)構(gòu)，我們必須為訓(xùn)練和測試數(shù)據(jù)創(chuàng)建兩個目錄，每個包含圖像文件夾和匹配的標(biāo)簽文件夾，其中每個xml文件除了擴(kuò)展名不同之外，都具有相同的文件名。而其余部分實際上是遵循一般的做法來訓(xùn)練Mask-RCNN的頂層，小伙伴可以在Matterport GitHub上查看不同的示例代碼。

我們必須創(chuàng)建一個類ParkingLot來加載數(shù)據(jù)集，通過解析帶注釋的xml文件提取邊界框的輪廓。

創(chuàng)建停車場類

然后，我們需要指定配置類，基于我們希望模型訓(xùn)練的類，加載訓(xùn)練和測試數(shù)據(jù)集，使用“訓(xùn)練”模式加載模型，并開始訓(xùn)練。

停車配置類

模型將在每個時期后保存在日志文件夾中，以便在訓(xùn)練完成后，我們可以繼續(xù)加載模型以評估其性能。使用Mask-RCNN實現(xiàn)這一點的方法是創(chuàng)建新的配置類，該類將限制我們的預(yù)測范圍。

預(yù)測配置類

用“驗證集”模式加載模型，從logs文件夾加載模型:

函數(shù)繪制檢測對象的邊界框

因此，如果我們加載任意隨機(jī)圖像，檢測可用和已占用的停車位，使用我們的函數(shù)來繪制它們，結(jié)果將是:

綠色框是可用的停車位，藍(lán)色邊界框是已占用的停車位。這個模型的性能看起來是相當(dāng)不錯，盡管它仍然不能檢測到小車/停車位。數(shù)據(jù)集中故意忽略了停在人行道旁的汽車，這樣模型在訓(xùn)練時就不會考慮這些汽車。這是MaskRCNN可視化方法的結(jié)果:

Mask-RCNN檢測可用停車位

考慮到這個結(jié)果，似乎模型的性能真的很好。但在我們的數(shù)據(jù)集中，我們只有停車場的兩個角度，因為我只訓(xùn)練頂級模型，我只使用了數(shù)據(jù)集的一部分。所以，可以看一下另一張停車場的照片，用類似的攝像頭定位，但不是這個數(shù)據(jù)集的一部分。

很明顯，這個模型與我們的停車場數(shù)據(jù)集過于匹配，這意味著它在同一個數(shù)據(jù)集上表現(xiàn)很好，但我沒有說訓(xùn)練，因為訓(xùn)練和測試照片實際上是位于停車場上不同汽車的相同照片，雖然技術(shù)上沒有泄漏測試數(shù)據(jù)，但在某種意義上，他們是一樣的。預(yù)先訓(xùn)練過的、開箱即用的Mask-RCNN在探測車輛方面的表現(xiàn)要好得多。

鑒于這些結(jié)論和我之前提到的局限性，為了改進(jìn)模型或向前推進(jìn)，下一步應(yīng)該做什么?我們可以使用整個數(shù)據(jù)集從頭開始訓(xùn)練模型，看看它的執(zhí)行情況，嘗試調(diào)優(yōu)檢測閾值。我們也可以試著去做，也許這可以成為這篇文章的第二部分。但理想情況下，更多變化的數(shù)據(jù)集是必不可少的，完整的注釋是創(chuàng)建準(zhǔn)確的停車位檢測模型的基礎(chǔ)。此外，為了建立不僅準(zhǔn)確而且健壯的檢測模型，考慮到正在行駛而尚未停車的汽車和擅自停放的汽車是非常重要的，這些可能是未來需要進(jìn)一步研究的步驟。


往期精彩回顧




適合初學(xué)者入門人工智能的路線及資料下載
機(jī)器學(xué)習(xí)及深度學(xué)習(xí)筆記等資料打印
機(jī)器學(xué)習(xí)在線手冊
深度學(xué)習(xí)筆記專輯
《統(tǒng)計學(xué)習(xí)方法》的代碼復(fù)現(xiàn)專輯
AI基礎(chǔ)下載
機(jī)器學(xué)習(xí)的數(shù)學(xué)基礎(chǔ)專輯
黃海廣老師《機(jī)器學(xué)習(xí)課程》課件合集
本站qq群851320808，加入微信群請掃碼：

国产秋霞理论久久久电影-婷婷色九月综合激情丁香-欧美在线观看乱妇视频-精品国avA久久久久久久-国产乱码精品一区二区三区亚洲人-欧美熟妇一区二区三区蜜桃视频

【深度學(xué)習(xí)】實戰(zhàn)：使用Mask-RCNN的停車位檢測