cszn / FFDNet

FFDNet: Toward a Fast and Flexible Solution for CNN based Image Denoising (TIP, 2018)
https://ieeexplore.ieee.org/abstract/document/8365806/
469 stars 130 forks source link

Training problem #28

Open rogerkuo1981 opened 3 years ago

rogerkuo1981 commented 3 years ago

I use the Train400 dataset for FFDNet-master\TrainingCodes\FFDNet_TrainingCodes_v1.0\Demo_Train_FFDNet_gray.m. The training looks no problem, outputs like below - train: epoch 499 : 1/ 4: loss: 2.3612 train: epoch 499 : 2/ 4: loss: 3.2255 train: epoch 499 : 3/ 4: loss: 3.5821 train: epoch 499 : 4/ 4: loss: 3.0635 train: epoch 500 : 1/ 4: loss: 3.9296 train: epoch 500 : 2/ 4: loss: 4.1511 train: epoch 500 : 3/ 4: loss: 2.4029 train: epoch 500 : 4/ 4: loss: 2.2997 But I did not see the final trained model. Could you please help to give some recommendation that what should I do?