HivisionIDPhotos : outil open source de création de photos d'identité intelligentes par l'IA

Introduction générale

HivisionIDPhotos est un outil de création de photos d'identité par IA, léger et open source, capable d'identifier intelligemment la scène photographique de l'utilisateur et de générer une photo d'identité standard répondant à diverses spécifications. L'outil prend en charge les couleurs et les tailles d'arrière-plan personnalisées et, à l'avenir, il introduira également des fonctions de beauté et de changement intelligent de la tenue vestimentaire. Les utilisateurs peuvent facilement créer un service local par le biais d'un déploiement Docker, adapté aux besoins de production de photos d'identité des particuliers et des entreprises.

Liste des fonctions

• Incrustation légère : utilisez des algorithmes d'IA efficaces pour incruster avec précision des portraits et générer des images avec des arrière-plans transparents.
• Génération de photos d'identité multi-tailles : prise en charge des photos de passeport, de visa et d'autres spécifications de la génération de photos d'identité.
• Génération de photos composées de six pouces : Composez plusieurs photos d'identité en une photo standard de six pouces pour faciliter l'impression et la production de lots.
• Changement intelligent de l'arrière-plan : permet d'ajouter ou de modifier la couleur de l'arrière-plan des photos d'identité.
• Service API : service API basé sur FastAPI, qui permet aux développeurs d'intégrer la fonction de photo d'identité.
• Prise en charge de Docker : mise en service rapide des services, simplification du processus d'installation et amélioration de la compatibilité des systèmes.
• Interaction en direct avec le web : le processus de génération d'une carte d'identité avec photo se déroule en ligne sur une page web locale fournie par Gradio Demo.

Utiliser l'aide

Installation et configuration de l'environnement

1. Projet de clonage :

   git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git
   cd HivisionIDPhotos
   

2. Installation d'environnements dépendants :

   pip install -r requirements.txt
   

3. Télécharger le fichier de modèle : placer le fichier de modèle pré-entraîné hivision_modnet.onnx Téléchargez-le et enregistrez-le dans le répertoire racine du projet.

Gestion des services locaux

1. Démarrer la démo Gradio :

   python app.py
   

   Après l'exécution du programme, une page web locale sera générée où l'utilisateur pourra terminer la génération et l'ajustement de la photo d'identité.

2. Déploiement des services API :

   python deploy_api.py
   

   En déployant des services API via FastAPI, les utilisateurs peuvent générer des photos d'identité et modifier l'arrière-plan via l'API.

Utiliser les services API

1. Envoyez une demande de création de photo d'identité :

   import requests
   
   url = "http://127.0.0.1:8080"
   files = {'file': open('test.jpg', 'rb')}
   response = requests.post(f"{url}/generate_idphoto", files=files)
   with open('idphoto.png', 'wb') as f:
       f.write(response.content)
   

2. Ajouter la couleur de base :

   import requests
   
   url = "http://127.0.0.1:8080"
   files = {'file': open('idphoto.png', 'rb')}
   data = {'color': '(0,0,0)'}
   response = requests.post(f"{url}/add_background", files=files, data=data)
   with open('idphoto_ab.jpg', 'wb') as f:
       f.write(response.content)
   

3. Créez une photo de 6 pouces de long, en format dactylographié :

   import requests
   
   url = "http://127.0.0.1:8080"
   files = {'file': open('idphoto_ab.jpg', 'rb')}
   response = requests.post(f"{url}/generate_layout_photos", files=files)
   with open('idphoto_layout.jpg', 'wb') as f:
       f.write(response.content)
   

Déploiement Docker

1. Construire l'image Docker :

   docker build -t hivisionidphotos .
   

2. Exécutez le conteneur Docker :

   docker run -p 8080:8080 hivisionidphotos
   

Expérience en ligne

https://huggingface.co/spaces/TheEeeeLin/HivisionIDPhotos

https://swanhub.co/ZeYiLin/HivisionIDPhotos/demo