하이비전ID사진: 오픈 소스 지능형 AI 사진 신분증 생성 도구

일반 소개

하이비전ID포토는 오픈 소스의 가벼운 AI 사진 신분증 생성 도구로, 사용자의 사진 장면을 지능적으로 식별하고 입력하여 다양한 사양을 충족하는 표준 신분증 사진을 생성할 수 있습니다. 이 도구는 사용자 지정 배경 색상과 크기를 지원하며, 향후에는 미용 및 지능형 정장 변경 기능도 도입할 예정입니다. 사용자는 도커 배포를 통해 개인 및 기업용 사진 ID 제작 요구에 적합한 로컬 서비스를 쉽게 구축할 수 있습니다.

기능 목록

• 경량 키잉: 효율적인 AI 알고리즘을 사용하여 인물 사진을 정확하게 키잉하고 투명한 배경의 이미지를 생성합니다.
• 다양한 크기의 신분증 사진 생성: 여권 사진, 비자 사진 및 기타 사양의 신분증 사진 생성을 지원합니다.
• 6인치 조판 사진 생성: 여러 장의 신분증 사진을 표준 6인치 사진으로 조판하여 간편하게 인쇄하고 일괄 제작할 수 있습니다.
• 지능형 배경 변경: 신분증 사진의 배경색을 추가하거나 변경할 수 있습니다.
• API 서비스: 개발자가 사진 ID 기능을 편리하게 통합할 수 있는 FastAPI 기반 API 서비스입니다.
• Docker 지원: 서비스를 빠르게 시작하고 실행하며, 설치 프로세스를 간소화하고, 시스템 호환성을 개선하세요.
• 실시간 웹 상호 작용: Gradio 데모에서 제공하는 로컬 웹 페이지를 통해 온라인으로 사진 신분증 생성 프로세스를 운영합니다.

도움말 사용

설치 및 환경 구성

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
   

   FastAPI를 통해 API 서비스를 배포하면 사용자는 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. 6인치 조판 사진을 생성합니다:

   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)
   

도커 배포

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