├── README.md ├── data ├── WFLW │ └── test.txt ├── demo.jpg ├── demo.jpg.bbox ├── demo.jpg.landmark ├── demo.jpg.png ├── demo2.jpg ├── demo2.jpg.bbox ├── demo2.jpg.landmark ├── demo2.jpg.png ├── figure.png ├── selfie.jpg ├── selfie.jpg.bbox ├── selfie.jpg.landmark └── selfie.jpg.png ├── examples ├── demo.py ├── test_300w.py ├── test_aflw2000.py ├── test_lapa.py ├── test_ls3dw.py ├── test_wflw.py └── thop-profile.py ├── models ├── 300w │ ├── hrnet18_256x256_p1 │ │ ├── config.yaml │ │ └── model.pth │ ├── hrnet18_256x256_p2 │ │ ├── config.yaml │ │ └── model.pth │ ├── mobilenet_128x128_p2 │ │ ├── config.yaml │ │ └── model.pth │ └── mobilenet_256x256_p2 │ │ ├── config.yaml │ │ └── model.pth ├── 300wlp │ └── pts3d │ │ ├── hrnet18_128x128_p2 │ │ ├── config.yaml │ │ └── model.pth │ │ └── mobilenet_128x128_p2 │ │ ├── config.yaml │ │ └── model.pth ├── README.md ├── lapa │ ├── hrnet18_256x256_p2 │ │ ├── config.yaml │ │ └── model.pth │ ├── hrnet18_256x256_p2_jd │ │ ├── config.yaml │ │ └── model.pth │ ├── hrnet32_256x256_p2 │ │ ├── config.yaml │ │ └── model.pth │ ├── mobilenet_128x128_p2 │ │ ├── config.yaml │ │ └── model.pth │ └── mobilenet_256x256_p2 │ │ ├── config.yaml │ │ └── model.pth └── wflw │ ├── hrnet18_256x256_p1 │ ├── config.yaml │ └── model.pth │ ├── hrnet18_256x256_p2 │ ├── config.yaml │ └── model.pth │ ├── mobilenet_128x128_p2 │ ├── config.yaml │ └── model.pth │ ├── mobilenet_160x160_p2 │ ├── config.yaml │ └── model.pth │ └── mobilenet_256x256_p2 │ ├── config.yaml │ └── model.pth ├── requirements.txt ├── setup.py └── torchalign ├── __init__.py ├── api.py ├── backbone ├── __init__.py ├── hrnet.py └── mobilenet.py ├── cfg.py └── heatmap_head ├── __init__.py ├── blocks ├── __init__.py └── head_block.py ├── heatmap_head.py └── transforms ├── __init__.py ├── functional.py └── module.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/README.md -------------------------------------------------------------------------------- /data/WFLW/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/WFLW/test.txt -------------------------------------------------------------------------------- /data/demo.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/demo.jpg -------------------------------------------------------------------------------- /data/demo.jpg.bbox: -------------------------------------------------------------------------------- 1 | 58 35 198 231 1.0 -------------------------------------------------------------------------------- /data/demo.jpg.landmark: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/demo.jpg.landmark -------------------------------------------------------------------------------- /data/demo.jpg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/demo.jpg.png -------------------------------------------------------------------------------- /data/demo2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/demo2.jpg -------------------------------------------------------------------------------- /data/demo2.jpg.bbox: -------------------------------------------------------------------------------- 1 | 306 308 696 870 1.0 -------------------------------------------------------------------------------- /data/demo2.jpg.landmark: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/demo2.jpg.landmark -------------------------------------------------------------------------------- /data/demo2.jpg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/demo2.jpg.png -------------------------------------------------------------------------------- /data/figure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/figure.png -------------------------------------------------------------------------------- /data/selfie.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/selfie.jpg -------------------------------------------------------------------------------- /data/selfie.jpg.bbox: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/selfie.jpg.bbox -------------------------------------------------------------------------------- /data/selfie.jpg.landmark: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/selfie.jpg.landmark -------------------------------------------------------------------------------- /data/selfie.jpg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/data/selfie.jpg.png -------------------------------------------------------------------------------- /examples/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/examples/demo.py -------------------------------------------------------------------------------- /examples/test_300w.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/examples/test_300w.py -------------------------------------------------------------------------------- /examples/test_aflw2000.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/examples/test_aflw2000.py -------------------------------------------------------------------------------- /examples/test_lapa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/examples/test_lapa.py -------------------------------------------------------------------------------- /examples/test_ls3dw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/examples/test_ls3dw.py -------------------------------------------------------------------------------- /examples/test_wflw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/examples/test_wflw.py -------------------------------------------------------------------------------- /examples/thop-profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/examples/thop-profile.py -------------------------------------------------------------------------------- /models/300w/hrnet18_256x256_p1/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300w/hrnet18_256x256_p1/config.yaml -------------------------------------------------------------------------------- /models/300w/hrnet18_256x256_p1/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300w/hrnet18_256x256_p1/model.pth -------------------------------------------------------------------------------- /models/300w/hrnet18_256x256_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300w/hrnet18_256x256_p2/config.yaml -------------------------------------------------------------------------------- /models/300w/hrnet18_256x256_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300w/hrnet18_256x256_p2/model.pth -------------------------------------------------------------------------------- /models/300w/mobilenet_128x128_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300w/mobilenet_128x128_p2/config.yaml -------------------------------------------------------------------------------- /models/300w/mobilenet_128x128_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300w/mobilenet_128x128_p2/model.pth -------------------------------------------------------------------------------- /models/300w/mobilenet_256x256_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300w/mobilenet_256x256_p2/config.yaml -------------------------------------------------------------------------------- /models/300w/mobilenet_256x256_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300w/mobilenet_256x256_p2/model.pth -------------------------------------------------------------------------------- /models/300wlp/pts3d/hrnet18_128x128_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300wlp/pts3d/hrnet18_128x128_p2/config.yaml -------------------------------------------------------------------------------- /models/300wlp/pts3d/hrnet18_128x128_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300wlp/pts3d/hrnet18_128x128_p2/model.pth -------------------------------------------------------------------------------- /models/300wlp/pts3d/mobilenet_128x128_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300wlp/pts3d/mobilenet_128x128_p2/config.yaml -------------------------------------------------------------------------------- /models/300wlp/pts3d/mobilenet_128x128_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/300wlp/pts3d/mobilenet_128x128_p2/model.pth -------------------------------------------------------------------------------- /models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/README.md -------------------------------------------------------------------------------- /models/lapa/hrnet18_256x256_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/hrnet18_256x256_p2/config.yaml -------------------------------------------------------------------------------- /models/lapa/hrnet18_256x256_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/hrnet18_256x256_p2/model.pth -------------------------------------------------------------------------------- /models/lapa/hrnet18_256x256_p2_jd/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/hrnet18_256x256_p2_jd/config.yaml -------------------------------------------------------------------------------- /models/lapa/hrnet18_256x256_p2_jd/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/hrnet18_256x256_p2_jd/model.pth -------------------------------------------------------------------------------- /models/lapa/hrnet32_256x256_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/hrnet32_256x256_p2/config.yaml -------------------------------------------------------------------------------- /models/lapa/hrnet32_256x256_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/hrnet32_256x256_p2/model.pth -------------------------------------------------------------------------------- /models/lapa/mobilenet_128x128_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/mobilenet_128x128_p2/config.yaml -------------------------------------------------------------------------------- /models/lapa/mobilenet_128x128_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/mobilenet_128x128_p2/model.pth -------------------------------------------------------------------------------- /models/lapa/mobilenet_256x256_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/mobilenet_256x256_p2/config.yaml -------------------------------------------------------------------------------- /models/lapa/mobilenet_256x256_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/lapa/mobilenet_256x256_p2/model.pth -------------------------------------------------------------------------------- /models/wflw/hrnet18_256x256_p1/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/hrnet18_256x256_p1/config.yaml -------------------------------------------------------------------------------- /models/wflw/hrnet18_256x256_p1/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/hrnet18_256x256_p1/model.pth -------------------------------------------------------------------------------- /models/wflw/hrnet18_256x256_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/hrnet18_256x256_p2/config.yaml -------------------------------------------------------------------------------- /models/wflw/hrnet18_256x256_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/hrnet18_256x256_p2/model.pth -------------------------------------------------------------------------------- /models/wflw/mobilenet_128x128_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/mobilenet_128x128_p2/config.yaml -------------------------------------------------------------------------------- /models/wflw/mobilenet_128x128_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/mobilenet_128x128_p2/model.pth -------------------------------------------------------------------------------- /models/wflw/mobilenet_160x160_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/mobilenet_160x160_p2/config.yaml -------------------------------------------------------------------------------- /models/wflw/mobilenet_160x160_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/mobilenet_160x160_p2/model.pth -------------------------------------------------------------------------------- /models/wflw/mobilenet_256x256_p2/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/mobilenet_256x256_p2/config.yaml -------------------------------------------------------------------------------- /models/wflw/mobilenet_256x256_p2/model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/models/wflw/mobilenet_256x256_p2/model.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/setup.py -------------------------------------------------------------------------------- /torchalign/__init__.py: -------------------------------------------------------------------------------- 1 | from .api import * -------------------------------------------------------------------------------- /torchalign/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/api.py -------------------------------------------------------------------------------- /torchalign/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/backbone/__init__.py -------------------------------------------------------------------------------- /torchalign/backbone/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/backbone/hrnet.py -------------------------------------------------------------------------------- /torchalign/backbone/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/backbone/mobilenet.py -------------------------------------------------------------------------------- /torchalign/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/cfg.py -------------------------------------------------------------------------------- /torchalign/heatmap_head/__init__.py: -------------------------------------------------------------------------------- 1 | from .heatmap_head import * -------------------------------------------------------------------------------- /torchalign/heatmap_head/blocks/__init__.py: -------------------------------------------------------------------------------- 1 | from .head_block import * -------------------------------------------------------------------------------- /torchalign/heatmap_head/blocks/head_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/heatmap_head/blocks/head_block.py -------------------------------------------------------------------------------- /torchalign/heatmap_head/heatmap_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/heatmap_head/heatmap_head.py -------------------------------------------------------------------------------- /torchalign/heatmap_head/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/heatmap_head/transforms/__init__.py -------------------------------------------------------------------------------- /torchalign/heatmap_head/transforms/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/heatmap_head/transforms/functional.py -------------------------------------------------------------------------------- /torchalign/heatmap_head/transforms/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/baoshengyu/H3R/HEAD/torchalign/heatmap_head/transforms/module.py --------------------------------------------------------------------------------