Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Do you have the generator weights? #10

Open
adar-cohen-imagenai opened this issue Oct 5, 2023 · 3 comments
Open

Do you have the generator weights? #10

adar-cohen-imagenai opened this issue Oct 5, 2023 · 3 comments

Comments

@adar-cohen-imagenai
Copy link

Hey, thank you for this project, well done!
Do you mind sharing your generator checkpoint (those that were trained based on the MDF loss)?
I want to simply run inference on the JPEG artifacts removal and see the results

@aamir-mustafa
Copy link
Collaborator

Hi, Thanks for your interest in our work. Please note that the discriminator checkpoints are used as feature extractors for training the jpeg artefacts removal model, instead of conventional loss functions. The underlying model does not change.

Hope that helps.

@adar-cohen-imagenai
Copy link
Author

@aamir-mustafa Hey Amir, thanks for your reply!
I totally understand, what I would like, is to have the generator model checkpoint, the one you train based on the MDF loss.

In your paper, you show examples of a few outputs, each output is some model based on some loss function. The model that was trained based on your MDF loss, reached the highest results. Im interested in that checkpoint of the generator, and not the discriminatory that calc the MDF loss.

@delyan-boychev
Copy link

@adar-cohen-imagenai Hello, Adar! I have come up with the same issue. You can check out the original repository of SinGAN - https://github.com/tamarott/SinGAN. The implementation of the generator and the discriminator are the same. You can change the perturbations added to the input in the file SinGAN/functions.py. After that, you can reproduce the results using the proposed perturbations in the paper.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants