HivisionIDPhotos: herramienta de código abierto para la creación de documentos de identidad inteligentes con IA

Introducción general

HivisionIDPhotos es una herramienta de código abierto y ligera para la creación de identificaciones fotográficas con IA que puede identificar de forma inteligente la escena fotográfica del usuario y teclearla para generar una foto de identificación estándar que cumpla una serie de especificaciones. La herramienta admite colores y tamaños de fondo personalizados, y en el futuro también introducirá funciones de belleza y cambio inteligente de vestimenta formal. Los usuarios pueden crear fácilmente un servicio local mediante el despliegue de Docker, adecuado para las necesidades de producción de documentos de identidad con foto personales y empresariales.

Lista de funciones

• Incrustación ligera: utilice algoritmos de IA eficaces para incrustar retratos con precisión y generar imágenes con fondos transparentes.
• Generación de fotos de carné de varios tamaños: admite foto de pasaporte, foto de visado y otras especificaciones de la generación de fotos de carné.
• Generación de fotos tipográficas de seis pulgadas: tipifique varias fotos de carné en una foto estándar de seis pulgadas para facilitar la impresión y la producción por lotes.
• Cambio de fondo inteligente: permite añadir o cambiar el color de fondo de las fotos de carné.
• Servicio API: servicio API basado en FastAPI, que resulta cómodo para que los desarrolladores integren la función de identificación con fotografía.
• Compatibilidad con Docker: ponga en marcha rápidamente los servicios, simplifique el proceso de instalación y mejore la compatibilidad del sistema.
• Interacción Web en Directo: Opere el proceso de generación del documento de identidad con fotografía en línea a través de una página web local proporcionada por la Demo de Gradio.

Utilizar la ayuda

Instalación y configuración del entorno

1. Proyecto de clonación:

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

2. Instalación de entornos dependientes:

   pip install -r requirements.txt
   

3. Descargar el archivo del modelo: Coloque el archivo del modelo preentrenado hivision_modnet.onnx Descárguelo y guárdelo en el directorio raíz del proyecto.

Funcionamiento de los servicios locales

1. Inicie la demostración de Gradio:

   python app.py
   

   Tras ejecutar el programa, se generará una página web local en la que el usuario podrá completar la generación y el ajuste de la foto de carné.

2. Despliegue de servicios API:

   python deploy_api.py
   

   Desplegando servicios API a través de FastAPI, los usuarios pueden generar fotos de carné y modificar el fondo a través de la API.

Uso de los servicios API

1. Enviar una solicitud de creación de foto de carné:

   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. Añadir el color 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. Genera una foto tipográfica de 15 cm:

   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)
   

Despliegue de Docker

1. Cree la imagen Docker:

   docker build -t hivisionidphotos .
   

2. Ejecute el contenedor Docker:

   docker run -p 8080:8080 hivisionidphotos
   

Experiencia en línea

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

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