HivisionIDPhotos: интеллектуальный инструмент для создания фотоидентификаторов с открытым исходным кодом

Общее введение

HivisionIDPhotos - это легкий инструмент для создания ID-фотографий с открытым исходным кодом, способный интеллектуально определять фотосцену пользователя и генерировать стандартную ID-фотографию, соответствующую различным спецификациям. Инструмент поддерживает пользовательские цвета и размеры фона, а в будущем он также представит функции красоты и интеллектуального изменения формальной одежды. Пользователи могут легко создать локальный сервис с помощью развертывания Docker, подходящий для производства персональных и корпоративных удостоверений личности.

Список функций

• Облегченный кеинг: используйте эффективные алгоритмы искусственного интеллекта для точного кеинга портретов и создания изображений с прозрачным фоном.
• Генерация многоразмерных фото ID: поддержка фото для паспорта, фото для визы и других спецификаций генерации фото ID.
• Создание шестидюймовых фотографий: напечатайте несколько фотографий для удостоверений личности на стандартной шестидюймовой фотографии для удобства печати и серийного производства.
• Интеллектуальная смена фона: поддержка добавления или изменения цвета фона для ID-фотографий.
• API-сервис: API-сервис на основе FastAPI, который удобен для разработчиков для интеграции функции фотоидентификации.
• Поддержка Docker: быстрый запуск служб, упрощение процесса установки и повышение совместимости систем.
• Живое взаимодействие с Интернетом: управляйте процессом создания удостоверения личности с фотографией в режиме онлайн с помощью локальной веб-страницы, предоставляемой Gradio Demo.

Использование помощи

Установка и настройка среды

1. Проект клонирования:

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

2. Установка зависимых сред:

   pip install -r requirements.txt
   

3. Загрузить файл модели: поместите файл предварительно обученной модели hivision_modnet.onnx Скачайте и сохраните его в корневом каталоге проекта.

Работа местных служб

1. Запустите демонстрационную версию Gradio:

   python app.py
   

   После запуска программы будет создана локальная веб-страница, на которой пользователь сможет завершить процесс создания и корректировки фотографии удостоверения личности.

2. Развертывание служб API:

   python deploy_api.py
   

   Развертывая API-сервисы через FastAPI, пользователи могут выполнять генерацию ID-фотографии, изменение фона и другие операции через API.

Использование служб API

1. Отправьте запрос на создание фотографии для удостоверения личности:

   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. Добавьте базовый цвет:

   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. Создайте шестидюймовую напечатанную фотографию:

   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)
   

Развертывание Docker

1. Создайте образ Docker:

   docker build -t hivisionidphotos .
   

2. Запустите контейнер Docker:

   docker run -p 8080:8080 hivisionidphotos
   

Опыт работы в Интернете

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

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