Makefile

.PHONY: all clean train_initial_states train_warp install-deps nuke package

all: train_initial_states train_warp

install-deps:
	@pip install -r requirements.txt
	@pip install -e .

package: clean
	@tar -cjf dist.tbz ifmorph/ warp-train.py warp-inference-vid.py warp-inference-image.py setup.py requirements.txt README.md LICENSE create-initial-states.py align.py Makefile download_data.sh
	@echo "Package created at: dist.tbz"

clean:
	@rm -Rf __pycache__
	@rm -Rf ifmorph/__pycache__

nuke: clean
	@rm -Rf pretrained
	@rm -Rf results
	@rm -Rf data/frll*

train_initial_states: data/frll_neutral_front
	@python create-initial-states.py --nsteps 5000 --device cuda:0 experiments/initial_state_rgb_large_im.yaml data/frll_neutral_front/001_03.jpg data/frll_neutral_front/002_03.jpg

train_warp: results/001_002-baseline/weights.pth
	@echo "FRLL 001->002 warp trained"

results/001_002-baseline/weights.pth: pretrained/frll_neutral_front
	@python warp-train.py --no-ui experiments/001_002-baseline.yaml

landmark_models/shape_predictor_68_face_landmarks_GTX.dat:
	@echo "Downloading DLib GTX 68 landmarks detection model"
	@curl --location --remote-header-name --remote-name https://github.com/davisking/dlib-models/raw/master/shape_predictor_68_face_landmarks_GTX.dat.bz2
	@bunzip2 shape_predictor_68_face_landmarks_GTX.dat.bz2
	@mkdir -p landmark_models
	@mv shape_predictor_68_face_landmarks_GTX.dat landmark_models/
	@rm -Rf shape_predictor_68_face_landmarks_GTX.dat.bz2
	@echo "Done"

data/frll_neutral_front:
	@echo "Downloading the FRLL dataset"
	@curl --location --remote-header-name --remote-name https://figshare.com/ndownloader/files/8541961
	@unzip neutral_front.zip
	@mkdir -p $@
	@mv neutral_front/* $@
	@rm -Rf __MACOSX neutral_front.zip neutral_front/
	@echo "Dataset downloaded"

pretrained/frll_neutral_front:
	@mkdir -p $@
	@python standalone/download-data.py 1QYoprK2bycXHItSkx9H8JfMGz48B9a3N frll_neutral_front.tar.bz2
	@tar -xjf frll_neutral_front.tar.bz2 || (echo "bzip failed $$?"; rm -Rf $@ cookie frll_neutral_front.tar.bz2; exit 1)
	@mv frll_neutral_front/*.pth $@
	@rm -Rf frll_neutral_front.tar.bz2 frll_neutral_front/

data/frll_neutral_front_cropped: data/frll_neutral_front
	@python align.py --just-crop --output-size 1350 --n-tasks 4 $< $@

pretrained/frll_neutral_front_cropped:
	@mkdir -p $@
	@python standalone/download-data.py 1guMg5ablWDQgaSfr5sFwScPWa-gm5Vsz frll_cropped.tar.bz2
	@tar -xjf frll_cropped.tar.bz2 || (echo "bzip failed $$?"; rm -Rf $@ cookie frll_cropped.tar.bz2; exit 1)
	@mv frll_cropped/*.pth $@
	@rm -Rf frll_cropped.tar.bz2 frll_cropped/