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

How to test trained stage 1 and stage 2 model? #144

Open
ElTocas opened this issue Dec 11, 2024 · 4 comments
Open

How to test trained stage 1 and stage 2 model? #144

ElTocas opened this issue Dec 11, 2024 · 4 comments

Comments

@ElTocas
Copy link

ElTocas commented Dec 11, 2024

I have trained model in specific lr images.

How can i infer the models .prh generated?

Thanks

@0x3f3f3f3fun
Copy link
Collaborator

I will add the code for self-trained model inference today.

@0x3f3f3f3fun
Copy link
Collaborator

Use this command for custom-trained model inference :)

Since DiffBIR needs to support multiple tasks, the inference code may appear somewhat complex. Feel free to ask if you have any questions about the code.

@ElTocas
Copy link
Author

ElTocas commented Dec 12, 2024

First of all, thanks for all the work you have done.

I did the process, and the result is a very black image. I just tested it for 1000 iter on 20 train images, but the result is all black.

Is there something that i missing?

ANJI_fig112 (5)

@0x3f3f3f3fun
Copy link
Collaborator

Can you provide more detailed information?,
such as: (1) your training configuration file; (2) whether your training logs are normal; (3) your inference command.

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

2 participants