1. 程式人生 > >SRCNN程式碼理解

SRCNN程式碼理解

先附上程式碼地址     https://github.com/tegg89/SRCNN-Tensorflow 

以灰度影象為例

glob.glob得到所有訓練集的圖片

先取出3的整數倍的畫素值, 以進行下面的影象模糊過程

input_ = scipy.ndimage.interpolation.zoom(label_, (1./scale), prefilter=False)

  input_ = scipy.ndimage.interpolation.zoom(input_, (scale/1.), prefilter=False)

通過 把圖片先縮小 再放大 ,得到模糊影象(後面要用的input),原始影象就是label,然後把所有的影象裁剪成33*33,和21*21 分別新增到兩個list中, 把兩個list分別轉成array,得到n*33*33*1 和n*21*21*1,然後存成.h5檔案格式,再讀出

注意 只按照塊作為一個訓練資料,不是一個影象作為一個數據

網路部分比較簡單,不再贅述