Avatar generator
- #Entertainment
The solution for creating artistic avatars for social media using images as input.
- Image Generation
- Machine Learning
Impact
The goal was to develop a solution that would allow users to generate a creative avatar from the uploaded photo for their social media accounts. The solution has several predefined presets while also allowing the user to tweak the image generation prompts.
Services we provided
The solution for creating artistic avatars for social media using images as input.
Tech Stack
Python
Huggingface
Flask
Stable Diffusion
Challenges and Solutions
🧐 Challenges
- Using a picture of the user to guide the image generation process.
- Developing image generation pipeline.
- Developing and testing presets for generating creative avatars.
💡 Solutions
Stable diffusion with a control net was used to create images based on pictures of the users. The avatar generation pipeline has the following steps:
- Extracting features from the user’s picture (edges, facial landmarks).
- Using these featuresa along with the prompt and appropriate stable diffusion model to generate a picture.
User flow
1. The user selects the style of the avatar (realistic, black and white, cartoon, painting, anime).
2. The appropriate stable diffusion model is chosen, and the proper text prompt is added.
3. The user can select from a list of presets or input their own prompt.
4. The user can upload their photo if they want the generated image to match their likeness, or they can omit to do it.
5. The model generates a batch of several images and presents them to the user.