昨天以及前天這兩個實現都是基於經典的 SRCNN 結構上,包含三層卷積層。
PyTorch 與 TensorFlow 實現說明
數據處理:
在兩種實現中,我們都將圖像轉換為 YCbCr 色彩空間,並只對亮度通道進行操作(丫通道),因為我們人類對亮度的變化更敏感,而Cb 和Cr通道則可以用簡單的插值方法處理。
圖像首先被載入,然後轉換為張量形式,這是神經網絡所需的輸入格式。
模型訓練:
•這兩個實現都使用了均方誤差 (MSE) 作為損失函數,這是超分辨率問題中常用的衡量標準。MSE 會衡量重建圖像與目標圖像之間的像素差異,並指導網絡的參數更新。
•訓練過程中,我們逐步更新網絡參數,使輸出的超分辩率圖像能夠逐漸接近目標的高解析度圖像。