數(shù)據(jù)集

VOC2007數(shù)據(jù)集有20個類：aeroplane, bicycle, bird, boat, bottle, bus, car, cat, chair, cow, diningtable, dog, horse, motorbike, person, pottedplant, sheep, sofa, train, tv/monitor。

VOC數(shù)據(jù)集下載地址如下，里面已經(jīng)包括了訓練集、測試集、驗證集（與測試集一樣），無需再次劃分：?

鏈接: https://pan.baidu.com/s/19Mw2u_df_nBzsC2lg20fQA?

提取碼: j5ge

解壓出來之后，下面的目錄結(jié)構(gòu)

生成數(shù)據(jù)

拿到數(shù)據(jù)之后，我們還需要利用上面的信息，生成模型需要的形式。

首先我們需要在model_data目錄下創(chuàng)建一個類別文件voc_classes.txt

數(shù)據(jù)處理函數(shù)

通過上面代碼會生成一個2007_train.txt和2007_test.txt文件，里面包含包含了圖片的地址，和bbox框的坐標和類別。