- Gaze Estimation : Gaze360.pkl
- Face Landmark : spiga_wflw.pt
- Face Detector : Downloads run within code on first run
- Linux
- Python < 3.11
- PyTorch >= 1.10.1
- CUDA >= 11.0
Note
- This code has been tested for rtx 3060+ gpu.
- sort_tracker-py doesn't compatible with Python 3.11
pip install -r requirements.txt
When installing sort_tracker-py, use the following script if lap error occurs.
conda install -c conda-forge lap
python face_ld_ge/demo_aimmo_flask.py --snapshot [/path/to/gaze/estimation/model/ckpt]
python face_ld_ge/demo_image.py --snapshot [/path/to/gaze/estimation/model/ckpt] --snapshot_landmark [/path/to/face/landmark/model/ckpt] --input [/path/to/input/directory] --output [/path/to/output/directory]
python face_ld_ge/ai-assist.py --snapshot [/path/to/face/landmark/model/ckpt] --gpu [gpu_id] --input [/path/to/input/directory] --output [/path/to/output/directory]