Allgemeine Einführung
HivisionIDPhotos ist ein quelloffenes, leichtgewichtiges KI-Tool zur Erstellung von Ausweisfotos, das auf intelligente Weise die Fotoszene des Benutzers erkennen und ein Standard-Ausweisfoto erstellen kann, das eine Vielzahl von Spezifikationen erfüllt. Das Tool unterstützt benutzerdefinierte Hintergrundfarben und -größen und wird in Zukunft auch Funktionen für Schönheit und den intelligenten Wechsel der formellen Kleidung einführen. Benutzer können durch Docker-Bereitstellung problemlos einen lokalen Dienst erstellen, der für die Produktion von Foto-IDs für Privatpersonen und Unternehmen geeignet ist.
Funktionsliste
- Leichtes Keying: Nutzen Sie effiziente KI-Algorithmen, um Porträts präzise zu keyen und Bilder mit transparentem Hintergrund zu erzeugen.
- Multi-Size-ID-Foto Generation: Unterstützung Passfoto, Visum Foto und andere Spezifikationen der ID-Foto Generation.
- Six-inch Typeset Photo Generation: Setzen Sie mehrere ID-Fotos in ein Standard-Sechs-Zoll-Foto, um den Druck und die Serienproduktion zu erleichtern.
- Intelligenter Hintergrundwechsel: unterstützt das Hinzufügen oder Ändern der Hintergrundfarbe für ID-Fotos.
- API-Service: FastAPI-basierter API-Service, der es Entwicklern ermöglicht, die Foto-ID-Funktion zu integrieren.
- Docker-Unterstützung: schnelle Inbetriebnahme von Diensten, Vereinfachung des Installationsprozesses und Verbesserung der Systemkompatibilität.
- Live-Web-Interaktion: Führen Sie den Prozess der Foto-ID-Erstellung online über eine lokale Webseite durch, die von der Gradio-Demo bereitgestellt wird.
Hilfe verwenden
Installation und Konfiguration der Umgebung
- Projekt Klonen:
git klonen. https://github.com/Zeyi-Lin/HivisionIDPhotos.git cd HivisionIDFotos
- Installation von abhängigen Umgebungen:
pip install -r anforderungen.txt
- Herunterladen der Modelldatei: Platzieren Sie die vortrainierte Modelldatei
hivision_modnet.onnx
Laden Sie sie herunter und speichern Sie sie im Stammverzeichnis des Projekts.
Lokale Dienste betreiben
- Starten Sie die Gradio-Demo:
python app.py
Nach der Ausführung des Programms wird eine lokale Webseite generiert, auf der der Benutzer die Erstellung und Anpassung des ID-Fotos abschließen kann.
- Bereitstellung von API-Diensten:
python deploy_api.py
Durch die Bereitstellung von API-Diensten über FastAPI können Benutzer ID-Fotos erstellen und den Hintergrund über die API ändern.
Verwendung von API-Diensten
- Senden Sie eine Anfrage zur Erstellung eines ID-Fotos:
importieren Anfragen url = "http://127.0.0.1:8080" files = {Datei".: öffnen(test.jpg, 'rb')} response = requests.post(f"{url}/generate_idphoto", files=files) mit öffnen(idphoto.png', 'wb') als f. f.write(response.content)
- Fügen Sie die Grundfarbe hinzu:
importieren Anfragen url = "http://127.0.0.1:8080" files = {Datei".: öffnen(idphoto.png', 'rb')} Daten = {'Farbe': '(0,0,0)'} response = requests.post(f"{url}/add_background", files=files, data=data) mit öffnen('idphoto_ab.jpg', 'wb') als f. f.write(response.content)
- Erstellen Sie ein sechs Zoll großes Satzfoto:
importieren Anfragen url = "http://127.0.0.1:8080" files = {Datei".: öffnen('idphoto_ab.jpg', 'rb')} response = requests.post(f"{url}/layout_photos_generieren", files=files) mit öffnen('idphoto_layout.jpg', 'wb') als f. f.write(response.content)
Docker-Bereitstellung
- Erstellen Sie das Docker-Image:
docker build -t hivisionidphotos .
- Starten Sie den Docker-Container:
docker run -p 8080:8080 hivisionidphotos