python製作yolov2目標檢測標註資料集__python

YOLOv2-photo-mark 2017-11-7

網上關於製作yolo資料集的代碼有好幾份，之前看到的有imageLabel和yolo-mark，這兩個功能都比較完善，但是代碼比較複雜，後者還需要配置，且得到的資料需要再經過一次轉換，如果是像我這樣只需要做一類物體的標記或者需要製作的資料集較小的朋友，更適合用本代碼中的方法。
這個代碼是基於網上的一份代碼修改的，剛剛找了半天已經找不到出處了，如果知道的同學可以告知一下。
最初的代碼比較簡陋，運行起來也有一些問題，且輸出的格式不能直接用於yolov2的訓練,修改過後基本沒什麼問題。 使用方法

在代碼檔案所在目錄下建立images檔案夾、labels檔案夾，將需要做標記的圖片放到images檔案夾中。最後得到的標籤檔案會存放到labels檔案夾下。
代碼比較簡潔，因此功能也相對來說比較簡陋，目前只能對一類物體進行標記，如果需要製作多類的標籤資料，只能修改代碼67行後重複運行代碼。以後有時間會將代碼進行完善。
本代碼用到了opencv，利用opencv進行標記，需要提前配置好opencv庫（既然要用yolov2，opencv肯定還是要裝的）。
在命令列視窗下，輸入 python YOLOv2_photo_mark.py進入標記介面。
左鍵拖動進行繪製，當前圖片繪製完成後，點擊右鍵繪製下一張圖片，直到標記完成images目錄下的所有圖片。

代碼在github上：點擊