mrinal18 / introspection

Introspection [WIP]

tl;dr

Prerequirements

pip install -r ./requirements.txt

Image

# classification
PYTHONPATH=. python ./scripts/train_image.py

# classification + metric learning
PYTHONPATH=. python ./scripts/train_image.py --use-ml

Video

# go through data/scripts/readme.md
# classification
PYTHONPATH=. python ./scripts/train_video.py  # ~11GB GPU required
PYTHONPATH=. python ./scripts/train_video.py --freeze-encoder  # ~5GB GPU required

# classification + metric learning
PYTHONPATH=. python ./scripts/train_video.py --use-ml  # ~20GB GPU required
PYTHONPATH=. python ./scripts/train_video.py --freeze-encoder --use-ml  # ~15GB GPU required

Demo

• Training pipeline
• Captum visualization