1. 程式人生 > >目標檢測帶標籤樣本增廣工具

目標檢測帶標籤樣本增廣工具

本工具主要用於基於影象的目標檢測樣本增廣。源標註檔案為xml檔案,採用voc標準標註格式,如下

樣本增廣工具主要實現resize、rotate、crop、flip、mosaic的一個或多個組合變換,並生成變換後的影象及對應的xml標註檔案。同時還提供一個顯示功能,用於檢驗標註框的準確性。該工具適用於批量樣本生成。以下為使用示例:

源樣本

1.resize

支援指定放縮和比例放縮

2.rotate

支援任意角度旋轉,並使用背景圖填充邊緣。有指定旋轉角度模式和隨機旋轉模式。支援旋轉過程中放縮。

3.crop

支援'RANDOM_CROP','CENTER_CROP','FIVE_CROP'三種模式,'RANDOM_CROP'模式下可任意指定crop圖片數。

4.flip

支援水平、垂直、水平垂直3種翻轉模式

5.Mosaic

對源資料集中樣本進行隨機組合變換後生成拼圖樣本。特別適用於偽造小目標樣本

6show

顯示功能,自動分配不同顏色顯示不同目標類別