├── .gitignore ├── LICENSE ├── README.md ├── assets ├── in_video.mp4 ├── out_video_axis.mp4 └── out_video_cube.mp4 ├── data └── .gitkeep ├── detect.py ├── download.sh ├── evaluate.py ├── main.py ├── models ├── __init__.py ├── common.py ├── mobilenetv2.py ├── mobilenetv3.py ├── resnet.py └── scrfd.py ├── requirements.txt ├── utils ├── datasets.py ├── general.py └── loss.py └── weights ├── .gitkeep └── det_10g.onnx /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/README.md -------------------------------------------------------------------------------- /assets/in_video.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/assets/in_video.mp4 -------------------------------------------------------------------------------- /assets/out_video_axis.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/assets/out_video_axis.mp4 -------------------------------------------------------------------------------- /assets/out_video_cube.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/assets/out_video_cube.mp4 -------------------------------------------------------------------------------- /data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/detect.py -------------------------------------------------------------------------------- /download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/download.sh -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/evaluate.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/models/common.py -------------------------------------------------------------------------------- /models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/models/mobilenetv2.py -------------------------------------------------------------------------------- /models/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/models/mobilenetv3.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/scrfd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/models/scrfd.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/utils/datasets.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/utils/loss.py -------------------------------------------------------------------------------- /weights/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /weights/det_10g.onnx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yakhyo/head-pose-estimation/HEAD/weights/det_10g.onnx --------------------------------------------------------------------------------