SRCNN-Partial-Convolution

Use Partial Convolution to solve SRCNN’s pixel loss of picture

必備

• Tensorflow
• Scipy version > 0.18 ('mode' option from scipy.misc.imread function)
• h5py
• matplotlib
• numpy
• csv

使用

For training, python main.py --is_train True
For testing, python main.py --is_train False

結果

經過訓練15000次，達到效果如下：
Ground truth:
Bicubic Interpolated image:

訓練0次:

訓練15000次:

訓練時間以及使用電腦設備

訓練15000次之時間：2小時15分鐘9秒
CPU : i7-8750H 2.2GHz
RAM : 32GB
GPU : RTX2080。

參考

• tegg89/SRCNN-Tensorflow
• Use Partial Convolution to solve SRCNN’s pixel loss of picture