information about trained models
pariasm opened this issue · 1 comments
pariasm commented
Thank you for porting this work to pytorch. Could you provide more information about the trained models given in the folder toflow_models?
In particular, for the video denoising model (denoise.pkl):
- In the original work there are 3 types of noise considered (Gaussian 15, Gaussian 25 and mixed noise). Your denoise.pkl, on which type of noise was it trained?
- Is it the one trained by the original authors or the one you retrained?
- If it is the one you retrained, which trainset did you use, and for how many epochs.
Coldog2333 commented
Thanks for being interested in pytoflow.
As for your proposed three questions:
- I remember that it was trained on the dataset with mixed noise, but I am not sure because it has been done for almost half a year. If it is important to your project, you can check the pytoflow/generate_testing_sample/addnoise.m to figure out the answer. I used this matlab script to process the original Vimeo-90K at that time.
- All of these models are retrained.
- The experiment settings including the used trainset, the super-parameters (epochs, learning rate, etc.), the pipeline, are the same as the original paper. The only exception is that we use lr=3*1e-5 in interpolation task while the learning rate in the original paper is 3*1e-4.
If you have other questions about this repository, feel free to contact me.