1. 程式人生 > >caffe資料層資料增廣

caffe資料層資料增廣

本次希望實現的主要功能是希望在訓練過程中,能夠實時的隨機改變影象資料的飽和度、亮度和對比度,進而能夠達到資料增廣的目的,增加資料的多樣性,使得訓練得到的模型的泛化性更好。而如果要達到這樣的目的,就需要對data_transformer.cpp進行相應的程式碼更改。除此以外,因為希望在訓練過程中,只對負樣本進行資料增廣,而不改變正樣本。所以還需要對data_layer層進行相應更改。因為最初生成的key中已經記錄了樣本的正負屬性,所以可以在對key進行相應處理知道屬性是“正”or“負”。