How do I use the trained model file to remove noise from my photos?

yu4u / noise2noise

An unofficial and partial Keras implementation of "Noise2Noise: Learning Image Restoration without Clean Data"

MIT License

1.07k stars 237 forks source link

How do I use the trained model file to remove noise from my photos? #42

Closed ghost closed 4 years ago

ghost commented 4 years ago

I use this command. !python3 test_model.py --weight_file clean.hdf5 --image_dir cleanimage --output_dir Resdir But it won't export the original size image.

It generates three image combinations.

What should I do to export the noise-reduced image and keep the original size?

yu4u commented 4 years ago

You can modify the code to save denoised images.

https://github.com/yu4u/noise2noise/blob/master/test_model.py#L54