conda env create -f environment.yml
conda activate humanref
pip install git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch
pip install git+https://github.com/openai/CLIP.git
pip install git+https://github.com/huggingface/diffusers.git@ce5504934ac484fca39a1a5434ecfae09eabdf41
git clone https://github.com/NVlabs/nvdiffrast
cd nvdiffrast && pip install .
cd .. && rm -rf nvdiffrast

mkdir -p Weights/econ_weights && cd Weights
sh ./third_parties/ECON/fetch_data.sh
mv data/* econ_weights/
cd ..

cd Weights
git lfs install
git clone https://huggingface.co/Salesforce/blip2-opt-2.7b
git clone https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5
cd ..

mkdir -p Weights/LPIPS && cd Weights/LPIPS
wget https://download.pytorch.org/models/vgg16-397923af.pth
cd ../..

mkdir -p Weights/SCHP

python third_parties/ECON/run_ECON_smpl.py --in_dir='./data/image_000355.jpg' --out_dir='./data/Results_ECON'

python run.py --config configs/humanref.yaml --train --gpu 0 image_path='./data/Results_ECON/image_000355/econ/imgs_crop/image_000355_0_rgba.png'