卷積神經網絡的學習過程雖然不需要手工選取特征,可以實現對圖像特征的自動化學習,但是這種學習需要以大量的學習資料作為基礎,即數據集。然而目前物體識別領域已有的經典數據集,如imagenet和voc,數量龐大,物體種類繁多,對本文的研究并不具有針對性。因此為了針對性地提升識別效果,我們需要建立基于卡口圖像的車型數據庫。
參照標準圖像分類數據集,安防監控公司對數據集進行車輛區域和車輛類型的標注,其中,車輛區域用一個矩形框標定,包括左上角頂點的橫縱坐標以及右下角頂點的橫縱坐標;車輛類型分為小轎車和大客車兩大類, 一個圖像的標注結果。本文原始數據集中共標定了2800張圖像,其中2000張小轎車,800張大客車。本文選取1900張小轎車,700張大客車作為訓練集,余下100張小轎車,100張大貨車作為驗證集。由于兩類訓練數據并不均衡,且訓練集數據對于卷積神經網絡的訓練是遠遠不足的,因此在原有已標定數據集的基礎上,我們又進行了數據的增廣具體的數據增廣