📢 System Requirements: Both the official Python inference code and the ComfyUI workflow were tested on Ubuntu 20.04 with Python 3.10, PyTorch 2.5.1, and CUDA 12.1 on an NVIDIA A800 GPU. Before ...
SHeaP learns to predict head geometry (FLAME parameters) from a single image, by predicting and rendering 2D Gaussians. This repository contains code and models for the FLAME parameter inference only.