logo avatars pipeline

Référentiel du pipeline Avatars du logo

Explication du blog: médium

Aperçu

Ce référentiel contient le backend d'une application de diffusion stable, spécialement conçue pour générer des logos créatifs. L'application utilise la version 1.5 de diffusion stable intégrée à un modèle de contrôle énigré de contrôle pour transformer les logos standard en versions plus créatives.

Rationaliser le code frontal ici

Caractéristiques

• STABLE DIFFUSION 1.5 avec un filet de contrôle Canny : exploite le dernier en IA pour générer des conceptions de logo uniques.
• Implémentation FastAPI : Construit à l'aide de FastAPI pour des services backend efficaces et évolutifs.
• Encodage d'image Base64 : accepte et traite les images codées au format Base64, rationalisant le processus de gestion de l'image.
• Processus d'inférence sans couture : effectue l'inférence sur les images décodées et renvoie les logos améliorés de manière créative au format Base64.

Usage

• Le backend FastAPI accepte un objet «BasesDrequest» contenant tous les paramètres nécessaires.
• Les images doivent être envoyées sous forme de chaînes codées de base64. Ceux-ci sont ensuite décodés au backend pour le traitement.
• Après l'inférence, l'image finale est réencodée dans Base64 et retournée à l'utilisateur.

Google Colab Notebook

Le référentiel inclus est un ordinateur portable Google Colab pour le logo Avatars POC. Ce carnet interactif vous permet d'exécuter l'ensemble du processus et de voir les résultats en temps réel. Copiez simplement le lien de n'importe quelle image et exécutez-le. Cahier de colab

Référence de l'API

Vérifier le rythme cardiaque

  GET /heartbeat

Paramètre	Taper	Description
none	none	Pour vérifier si le serveur s'exécute

Générer une image

  POST /generate_image 

Paramètres

Nom	Taper	Description
prompt	string	Requis . L'invite pour générer l'image.
encoded_control_net_image	string	Requis . L'image nette de contrôle codée au format Base64.
control_type	string	Requis . Le type de contrôle utilisé dans la génération d'images.
height	int	Requis . La hauteur de l'image générée.
width	int	Requis . La largeur de l'image générée.
controlnet_conditioning_scale	float	L'échelle du conditionnement net de contrôle. La valeur par défaut est 1.0.
negative_prompt	string	Une chaîne spécifiant les invites négatives. La valeur par défaut est "déformé, NSFW, Blurr".
base_model	string	Le modèle de base utilisé pour la génération. La valeur par défaut est "Digiplay / juggernaut_final".
num_inference_steps	int	Le nombre d'étapes d'inférence. La valeur par défaut est 20.
guidance_scale	float	L'échelle d'orientation. La valeur par défaut est 0,6.
num_images_per_prompt	int	Le nombre d'images générées par invite. La valeur par défaut est 1.

{
  "prompt" : " Colorful, jungle surrounding, trees, natural, detailed, hd, 4k, best quality, extremely detailed " ,
  "generated_image_encoded" : " base64_encoded_generated_image_string "
}

Soutenez-moi: