A implementação do DiffPortrait360 requer um processo passo a passo para concluir a preparação do ambiente:
Preparação do hardware
- Deve usar GPU NVIDIA (suporte CUDA)
- Configuração mínima: 30 GB de memória de vídeo (gera 32 fps)
- Configuração recomendada: 80 GB de memória de vídeo (por exemplo, placa de vídeo A6000)
instalação do software
- Confirmação da versão CUDA (recomendada 12.2): passa
nvcc --versionsonda - Crie um ambiente Conda para o Python 3.9:
conda create -n diffportrait360 python=3.9conda activate diffportrait360
- Clonar o repositório de código:
git clone https://github.com/FreedomGu/DiffPortrait360.git - Instalar dependências:
- entrar em
diffportrait360_releasediretório (no disco rígido do computador) - realizar
pip install -r requirements.txt
- entrar em
Implementação do modelo
Faça o download dos três modelos pré-treinados (PANO_HEAD_MODEL/Head_Back_MODEL/Diff360_MODEL) do Hugging Face e modifique oinference.shOs parâmetros de caminho na seção Por fim, os parâmetros de caminho são passados através da funçãotorch.cuda.is_available()Verifique a disponibilidade da GPU.
Essa resposta foi extraída do artigoDiffPortrait360: gere visualizações de cabeça em 360 graus a partir de um único retratoO
Não pode ser reproduzido sem permissão:Ferramentas de produtividade de IA " Como instalar e configurar o ambiente de tempo de execução do DiffPortrait360?
Português do Brasil 
简体中文 
English 
日本語 
Deutsch