Introdução geral
O HivisionIDPhotos é uma ferramenta leve e de código aberto para a criação de fotos de identificação com IA, capaz de identificar de forma inteligente a cena da foto do usuário e de gerar uma foto de identificação padrão que atenda a uma variedade de especificações. A ferramenta é compatível com cores e tamanhos de fundo personalizados e, no futuro, também introduzirá recursos de beleza e mudança inteligente de trajes formais. Os usuários podem criar facilmente um serviço local por meio da implantação do Docker, adequado para as necessidades de produção de identificação com foto pessoal e empresarial.
Lista de funções
- Chaveamento leve: use algoritmos de IA eficientes para chavear retratos com precisão e gerar imagens com fundos transparentes.
- Geração de fotos de identificação em vários tamanhos: suporta fotos de passaporte, fotos de visto e outras especificações da geração de fotos de identificação.
- Geração de fotos tipificadas de seis polegadas: tipifique várias fotos de identificação em uma foto padrão de seis polegadas para facilitar a impressão e a produção em lote.
- Alteração inteligente do plano de fundo: suporte para adicionar ou alterar a cor do plano de fundo das fotos de identificação.
- Serviço de API: serviço de API baseado em FastAPI, que é conveniente para os desenvolvedores integrarem a função de identificação com foto.
- Suporte ao Docker: coloque os serviços em funcionamento rapidamente, simplifique o processo de instalação e melhore a compatibilidade do sistema.
- Interação com a Web ao vivo: Opere o processo de geração de identificação com foto on-line por meio de uma página da Web local fornecida pelo Gradio Demo.
Usando a Ajuda
Instalação e configuração do ambiente
- Projeto de clonagem:
git clone https://github.com/Zeyi-Lin/HivisionIDPhotos.git cd HivisionIDPhotos
- Instalação de ambientes dependentes:
pip install -r requirements.txt
- Download do arquivo do modelo: Coloque o arquivo do modelo pré-treinado
hivision_modnet.onnx
Faça o download e salve-o no diretório raiz do projeto.
Execução de serviços locais
- Inicie a demonstração do Gradio:
python app.py
Após a execução do programa, será gerada uma página da Web local onde o usuário poderá concluir a geração e o ajuste da foto de identificação.
- Implantação de serviços de API:
python deploy_api.py
Ao implementar serviços de API por meio da FastAPI, os usuários podem gerar fotos de identificação e modificar o plano de fundo por meio da API.
Uso de serviços de API
- Enviar uma solicitação para criação de foto de identificação:
importação solicitações url = "http://127.0.0.1:8080" arquivos = {'arquivo': aberto('test.jpg', 'rb')} response = requests.post(f"{url}/generate_idphoto", files=files) com aberto('idphoto.png', 'wb') como f. f.write(response.content)
- Adicione a cor de base:
importação solicitações url = "http://127.0.0.1:8080" arquivos = {'arquivo': aberto('idphoto.png', 'rb')} data = {'cor': '(0,0,0)'} response = requests.post(f"{url}/add_background", files=files, data=data) com aberto('idphoto_ab.jpg', 'wb') como f. f.write(response.content)
- Gere uma foto tipográfica de 15 cm:
importação solicitações url = "http://127.0.0.1:8080" arquivos = {'arquivo': aberto('idphoto_ab.jpg', 'rb')} response = requests.post(f"{url}/generate_layout_photos", files=files) com aberto('idphoto_layout.jpg', 'wb') como f. f.write(response.content)
Implementação do Docker
- Crie a imagem do Docker:
docker build -t hivisionidphotos .
- Execute o contêiner do Docker:
docker run -p 8080:8080 hivisionidphotos