YOLOv3訓(xùn)練自己的數(shù)據(jù)

關(guān)鍵點在于:弄清YOLOv3數(shù)據(jù)集的組織形式,需要按照這個生成數(shù)據(jù)。

一. 訓(xùn)練環(huán)境
1. YOLOv3編譯:OpenCV+GPU
git clone https://github.com/pjreddie/darknet
cd darknet
vim ...

2. 下載預(yù)訓(xùn)練權(quán)重(只包含卷積層)
    wget https://pjreddie.com/media/files/darknet53.conv.74 

二. 數(shù)據(jù)準備
0. 因為YOLOv3提供了一個將VOC數(shù)據(jù)集轉(zhuǎn)成自己訓(xùn)練的數(shù)據(jù)集格式的工具:scripts/voc_label.py,
因此,訓(xùn)練數(shù)據(jù)只需要按VOC數(shù)據(jù)集的格式準備就好, 再執(zhí)行voc_label.py即可
另一種思路,直接生成YOLO格式的?YOLO的格式更簡潔些。
1. VOC2012
http://host.robots.ox.ac.uk/pascal/VOC/voc2012/
研究下文件組織結(jié)構(gòu)
Annotations:用于存放與圖片對應(yīng)的XML文件
JPEGImages: 存放所有的圖片,png是否可以?
2. 對自己的圖片數(shù)據(jù)打標,按照YOLOv3的方式準備數(shù)據(jù)集
人工選取: l78z/projects/video_roi/video_roi_manual.py
自動保存: l78z/projects/video_roi/video_roi_crop.py
其他:labelImg: https://github.com/tzutalin/labelImg#macos
很多教程是基于次標注工具產(chǎn)出數(shù)據(jù),再轉(zhuǎn)成yolo訓(xùn)練數(shù)據(jù)
該工具支持直接生成YOLO格式,
基于cvui搞個工具?
ctrl+t , 打開新視頻
左鍵選擇,右鍵取消
3. 使用 scripts/voc_label.py 生成

三、 修改YOLOv3的配置文件
1. data/voc.names #修改成自己的分類
2. cfg/voc.data
classes = N #(N為自己的分類數(shù)量,如有10類不同的對象,N = 10)
train = /home/XXX/darknet/trainImagePath.txt # 訓(xùn)練集完整路徑列表
valid = /home/XXX/darknet/validateImagePath.txt # 測試集完整路徑列表
names = data/voc.names # 類別文件
backup = backup #(訓(xùn)練結(jié)果保存在darknet/backup/目錄下)
3. cfg/yolov3-voc.cfg
classes = N (N為自己的分類數(shù))
filters = 3*(classes+1+4), 修改每一個[yolo]層(一共有3處)之前的filters.
(可選) 修改訓(xùn)練的最大迭代次數(shù), max_batches = N

四、 YOLOv3訓(xùn)練&測試
訓(xùn)練:
./darknet detector train cfg/voc.data cfg/yolov3-voc.cfg darknet53.conv.74
測試:
./darknet detector test cfg/voc.data cfg/yolov3-voc.cfg backup/yolov3-voc_final.weights 01.jpg

五、題外話,說說視頻標注工具的設(shè)計
1. 讀取video_list.txt,創(chuàng)建目錄文件,下載視頻,存放至約定的文件夾,文件價格結(jié)構(gòu):
video_list.txt, video_id \t video_download_url
Videos/ 用于存放下載的視頻
Manual/ 用于存放用戶選擇的圖片、區(qū)域、區(qū)域裁剪圖
JPEGImages/ 存放自動生成的圖片
Annotations/ 用于圖片對應(yīng)的選取信息
class.txt 所有分類匯總
class/ 選取區(qū)域的圖像 ?
class1/file1.jpg
2. 自動定位至還沒有標注數(shù)據(jù)的,顯示標注進度。
3. 快捷鍵設(shè)定:按游戲的設(shè)定w前進幀,s后退幀,d下一個,a上一個(上下切換視頻,左右調(diào)節(jié)幀,方向鍵位于右側(cè),影響鼠標操作),上下切換時,自動保存。
4. 拖拽選中區(qū)域,要有x/y輔助線,右鍵取消上一次選中,
5. 選中結(jié)束,彈窗,類名,設(shè)置類名時,除了文字外有對應(yīng)的縮略圖
6. 保存VOC格式, 全部結(jié)束后,自動生成YOLO格式。

參考文檔:
https://blog.csdn.net/dcrmg/article/details/81296520
http://www.lxweimin.com/p/f4518fe04da1
http://www.lxweimin.com/p/91eafe0f3719
https://www.cnblogs.com/pprp/p/9525508.html

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,030評論 6 531
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,310評論 3 415
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 175,951評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,796評論 1 309
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,566評論 6 407
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,055評論 1 322
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,142評論 3 440
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 42,303評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 48,799評論 1 333
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 40,683評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,899評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,409評論 5 358
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 44,135評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,520評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,757評論 1 282
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,528評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,844評論 2 372

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