自編碼AE 實現圖片去馬賽克(完整代碼奉獻) pytorch

前言:博主閒來無事,寫了個AE網絡來實現任意圖片的去馬賽克(前提是訓練樣本是哪方面的馬賽克圖~~,博主是針對打碼人臉的去碼)。樣本分爲沒打碼人臉8000張和經打碼處理的對應8000張(取自celeba人臉數據集)。預處理階段:人臉resize到256x256,打碼。爲讀入數據,把文件夾裏的圖片名讀取出並保存到了label.txt文件。主程序分4塊:網絡定義文件net.py,  讀數據dataset
相關文章
相關標籤/搜索