kyustorm7 / eg3d_VideoCrop

Cropping Videos for EG3D Inversion

Code for Smooth Cropping for EG3D Inversion

Setup & Requirements

NVIDIA GPUs. We have done all testings on RTX 3090 GPU.

64-bit Python 3.9, PyTorch 1.11.0 + CUDA toolkit 11.3

cd Deep3DFaceRecon_pytorch
git clone https://github.com/NVlabs/nvdiffrast
cd nvdiffrast    # ./Deep3DFaceRecon_pytorch/nvdiffrast
pip install .

cd ..    # ./Deep3DFaceRecon_pytorch
git clone https://github.com/deepinsight/insightface.git
cp -r ./insightface/recognition/arcface_torch ./models/

Input Videos

mkdir -p frames
ffmpeg -i test.mp4  frames/frame_%04d.png   # Extract Frames from Video
python smooth_cropping.py --in_root frames