imgaug機(jī)器學(xué)習(xí)圖像增強(qiáng)
imgaug 是一個(gè)用于機(jī)器學(xué)習(xí)實(shí)驗(yàn)中圖像增強(qiáng)的庫(kù)。它支持廣泛的增強(qiáng)技術(shù),允許輕松組合這些技術(shù)并以隨機(jī)順序或在多個(gè) CPU 內(nèi)核上執(zhí)行它們;具有簡(jiǎn)單而強(qiáng)大的隨機(jī)界面,不僅可以增強(qiáng)圖像,還可以增強(qiáng)關(guān)鍵點(diǎn)/地標(biāo)、邊界框、熱圖和分割圖。
特性:
- 許多增強(qiáng)技術(shù)
- 例如仿射變換、透視變換、對(duì)比度變化、高斯噪聲、區(qū)域丟失、色調(diào)/飽和度變化、裁剪/填充、模糊……
- 為高性能而優(yōu)化
- 易于僅對(duì)某些圖像應(yīng)用增強(qiáng)
- 易于以隨機(jī)順序應(yīng)用增強(qiáng)
- 支持
- 圖像(完全支持 uint8,其他 dtypes 可參閱文檔)
- 熱圖 (float32)、分割圖 (int)、面具 (bool)
- 可能比相應(yīng)的圖像小/大。例如裁剪不需要額外的代碼行。
- 關(guān)鍵點(diǎn)/地標(biāo)(整數(shù)/浮點(diǎn)坐標(biāo))
- 邊界框(整數(shù)/浮點(diǎn)坐標(biāo))
- 多邊形(整數(shù)/浮點(diǎn)坐標(biāo))
- 線串(整數(shù)/浮點(diǎn)坐標(biāo))
- 采樣隨機(jī)值的自動(dòng)對(duì)齊
- 示例:通過(guò)從
uniform(-10°, 45°)中采樣的相同值旋轉(zhuǎn)圖像和分割圖。(0 extra lines of code)
- 示例:通過(guò)從
- 概率分布作為參數(shù)
- 示例:通過(guò)從
uniform(-10°, 45°)中采樣的值旋轉(zhuǎn)圖像。 - 示例:通過(guò)從
ABS(N(0, 20.0))*(1+B(1.0, 1.0))“中采樣的值旋轉(zhuǎn)圖像,其中ABS(.)是絕對(duì)函數(shù)、N(.)高斯分布和B(.)beta 分布。
- 示例:通過(guò)從
- 許多輔助功能
- 示例:繪制熱圖、分割圖、關(guān)鍵點(diǎn)、邊界框……
- 示例:縮放分割圖、圖像/地圖的平均/最大池、將圖像填充到縱橫比(例如對(duì)它們進(jìn)行平方)
- 示例:將關(guān)鍵點(diǎn)轉(zhuǎn)換為距離圖,從圖像中提取邊界框內(nèi)的像素,將多邊形裁剪到圖像平面,...
- 支持在多個(gè) CPU 內(nèi)核上進(jìn)行擴(kuò)充
評(píng)論
圖片
表情
