Keras實時增強增加噪聲和對比度

Question

Keras爲實時增強提供ImageDataGenerator類，但它不包括對比度調整和噪聲添加。

我們如何在訓練過程中應用隨機等級的噪音和隨機對比度調整？可以將這些函數添加到數據集中的'preprocessing_function'參數中嗎？

謝謝。

Answer 1

從Keras文檔：

preprocessing_function：函數，將每個輸入來暗示。該函數將在任何其他修改之前運行。函數應該有一個參數：一個圖像（等級爲3的Numpy張量），並且應該輸出具有相同形狀的Numpy張量。

因此，我創建了一個簡單的功能，然後使用imgaug module中的圖像增強功能。請注意，imgaug要求圖片排名爲4.