DnCNN-PyTorch
这是TIP2017论文的PyTorch实施,。 作者的。
这段代码是用PyTorch = 0.4。 移植代码很容易。 请参阅《 。
怎么跑
1.依存关系
(<0.4)
适用于Python的OpenCV
(用于PyTorch的TensorBoard)
2.训练DnCNN-S(已知噪声水平的DnCNN)
python train.py \
--preprocess True \
--num_of_layers 17 \
--mode S \
--noiseL 25 \
--val_noiseL 25
笔记
如果您已经构建了训练和验证数据集(即train.h5和val.h5文件),请将preprocess设置为False。
根据该论文,DnCNN-S具有17层。
noise
2021-04-25 17:29:53
23.62MB
Python
1