KAIR
KAIR copied to clipboard
FFDNET Training
I want to train the FFDNET model from the scratch for gray scale Images in Pytorch framework without using the pre-trained models available at https://drive.google.com/drive/folders/13kfr3qny7S2xwG9h7v95F5mkWs0OmU0D
After putting the dataset in the respective folders and set the parameters in train.jsonn file, I run main_train.py only but I noticed that there is no model saved in the model_ zoo folder while training which later I can use on my test Images to remove the AWGN noise from test data.
Can anyone guide me in this regard? Should I have to run other files as well in the training process(like network_ffdnet or util_model .py)
If anyone implement the ffdnet then kindly guide me. Thanks in Advance!