Die Bereitstellung von DiffPortrait360 erfordert einen schrittweisen Prozess zur Vorbereitung der Umgebung:
Vorbereitung der Hardware
- Muss NVIDIA GPU verwenden (CUDA-Unterstützung)
- Mindestkonfiguration: 30 GB Videospeicher (erzeugt 32 fps)
- Empfohlene Konfiguration: 80GB Videospeicher (z.B. A6000 Grafikkarte)
Software-Installation
- Bestätigung der CUDA Version (empfohlen 12.2): bestanden
nvcc --version
Sonde - Erstellen Sie eine Conda-Umgebung für Python 3.9:
conda create -n diffportrait360 python=3.9
conda activate diffportrait360
- Klonen Sie das Code-Repository:
git clone https://github.com/FreedomGu/DiffPortrait360.git
- Abhängigkeiten installieren:
- gehen in
diffportrait360_release
Verzeichnis (auf der Festplatte des Computers) - vollziehen
pip install -r requirements.txt
- gehen in
Einsatz des Modells
Laden Sie die drei vortrainierten Modelle (PANO_HEAD_MODEL/Head_Back_MODEL/Diff360_MODEL) von Hugging Face herunter und ändern Sie dieinference.sh
Die Pfadparameter in der Schließlich werden die Pfadparameter durch dietorch.cuda.is_available()
Überprüfen Sie die Verfügbarkeit von GPUs.
Diese Antwort stammt aus dem ArtikelDiffPortrait360: Erzeugen von 360-Grad-Kopfansichten aus einem einzigen PorträtDie