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

Training for specific use cases #20

Closed
LukaGiorgadze opened this issue Jan 23, 2024 · 3 comments
Closed

Training for specific use cases #20

LukaGiorgadze opened this issue Jan 23, 2024 · 3 comments

Comments

@LukaGiorgadze
Copy link

Hi, thank you for the great work. I want to create shadows for objects on a white background. I tried using a pre-trained model, but the results weren't great. Can I train a model specifically for this use case? If yes - what is your recommendations?
Thanks.

shadow_generation_result2 (1)
shadow_generation_result2
shadow_generation_result2 (3)

@ustcnewly
Copy link
Contributor

Our used training set DESOBAv2 does not contain such images with white backgrounds. To adapt to your problem, you are recommended to construct your own dataset and train ControlNet.

@LukaGiorgadze
Copy link
Author

LukaGiorgadze commented Jan 25, 2024

Our used training set DESOBAv2 does not contain such images with white backgrounds. To adapt to your problem, you are recommended to construct your own dataset and train ControlNet.

Thank you so much! I still use libcom for image composition and love it! Back to the original question - do you mean build my own dataset and train directly ControlNet? Is it enough? Do you have any other recommendations? Do I need to consider something else?

@ustcnewly
Copy link
Contributor

Thanks for your apprecitation! Yes, I mean that.You can try this simple method first and observe the results.

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