├── .gitignore ├── LICENSE ├── README.md ├── config.json ├── etc ├── mfw_mini.png └── pipeline.jpg ├── main.py ├── mask_functions.py ├── masks ├── mask001.png ├── mask001_01.png ├── mask002.png ├── mask003.png ├── mask004.png ├── mask004_01.png ├── mask005.png ├── mask006.png ├── mask007.png ├── mask007_01.png ├── mask008.png ├── mask009.png ├── mask010.png ├── mask011.png ├── mask012.png ├── mask013.png ├── mask014.png └── mask015.png ├── misc_utils.py ├── models ├── .gitignore ├── mobilenet_v1.py └── phase1_wpdc_vdc.pth.tar ├── nurbs_surf.mtl ├── obj_loader.py ├── requirements.txt ├── src ├── 00001_001.jpg ├── 00002_001.jpg ├── 00003_001.jpg ├── 00004_001.jpg ├── 00005_001.jpg ├── 00006_001.jpg ├── 00007_001.jpg ├── 00008_001.jpg ├── 00009_001.jpg ├── 00011_001.jpg ├── 00012_001.jpg ├── 00013_001.jpg ├── 00014_001.jpg ├── 00015_001.jpg ├── 00016_001.jpg ├── 00017_001.jpg ├── 00018_001.jpg ├── 00019_001.jpg ├── 00020_001.jpg ├── 00021_001.jpg ├── 00022_001.jpg ├── 00023_001.jpg ├── 00024_001.jpg ├── 00025_001.jpg ├── 00026_001.jpg ├── 00027_001.jpg ├── 00028_001.jpg ├── 00029_001.jpg ├── 00030_001.jpg ├── 00031_001.jpg ├── 00032_001.jpg ├── 00033_001.jpg ├── 00034_001.jpg ├── 00035_001.jpg ├── 00036_001.jpg ├── 00037_001.jpg ├── 00038_001.jpg ├── 00039_001.jpg ├── 00040_001.jpg ├── 00041_001.jpg ├── 00042_001.jpg ├── 00043_001.jpg ├── 00044_001.jpg ├── 00045_001.jpg ├── 00046_001.jpg ├── 00047_001.jpg ├── 00048_001.jpg ├── 00049_001.jpg └── 00050_001.jpg ├── train.configs ├── Model_PAF.pkl ├── keypoints_sim.npy ├── param_whitening.pkl ├── pncc_code.npy ├── u_exp.npy ├── u_shp.npy ├── w_exp_sim.npy └── w_shp_sim.npy └── utils ├── __init__.py ├── cv_plot.py ├── cython ├── __init__.py ├── mesh_core.cpp ├── mesh_core.h ├── mesh_core_cython.cpp ├── mesh_core_cython.pyx ├── readme.md └── setup.py ├── ddfa.py ├── estimate_pose.py ├── inference.py ├── io.py ├── lighting.py ├── paf.py ├── params.py └── render.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/README.md -------------------------------------------------------------------------------- /config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/config.json -------------------------------------------------------------------------------- /etc/mfw_mini.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/etc/mfw_mini.png -------------------------------------------------------------------------------- /etc/pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/etc/pipeline.jpg -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/main.py -------------------------------------------------------------------------------- /mask_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/mask_functions.py -------------------------------------------------------------------------------- /masks/mask001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask001.png -------------------------------------------------------------------------------- /masks/mask001_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask001_01.png -------------------------------------------------------------------------------- /masks/mask002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask002.png -------------------------------------------------------------------------------- /masks/mask003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask003.png -------------------------------------------------------------------------------- /masks/mask004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask004.png -------------------------------------------------------------------------------- /masks/mask004_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask004_01.png -------------------------------------------------------------------------------- /masks/mask005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask005.png -------------------------------------------------------------------------------- /masks/mask006.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask006.png -------------------------------------------------------------------------------- /masks/mask007.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask007.png -------------------------------------------------------------------------------- /masks/mask007_01.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask007_01.png -------------------------------------------------------------------------------- /masks/mask008.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask008.png -------------------------------------------------------------------------------- /masks/mask009.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask009.png -------------------------------------------------------------------------------- /masks/mask010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask010.png -------------------------------------------------------------------------------- /masks/mask011.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask011.png -------------------------------------------------------------------------------- /masks/mask012.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask012.png -------------------------------------------------------------------------------- /masks/mask013.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask013.png -------------------------------------------------------------------------------- /masks/mask014.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask014.png -------------------------------------------------------------------------------- /masks/mask015.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/masks/mask015.png -------------------------------------------------------------------------------- /misc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/misc_utils.py -------------------------------------------------------------------------------- /models/.gitignore: -------------------------------------------------------------------------------- 1 | shape_predictor_68_face_landmarks.dat 2 | -------------------------------------------------------------------------------- /models/mobilenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/models/mobilenet_v1.py -------------------------------------------------------------------------------- /models/phase1_wpdc_vdc.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/models/phase1_wpdc_vdc.pth.tar -------------------------------------------------------------------------------- /nurbs_surf.mtl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/nurbs_surf.mtl -------------------------------------------------------------------------------- /obj_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/obj_loader.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/00001_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00001_001.jpg -------------------------------------------------------------------------------- /src/00002_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00002_001.jpg -------------------------------------------------------------------------------- /src/00003_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00003_001.jpg -------------------------------------------------------------------------------- /src/00004_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00004_001.jpg -------------------------------------------------------------------------------- /src/00005_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00005_001.jpg -------------------------------------------------------------------------------- /src/00006_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00006_001.jpg -------------------------------------------------------------------------------- /src/00007_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00007_001.jpg -------------------------------------------------------------------------------- /src/00008_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00008_001.jpg -------------------------------------------------------------------------------- /src/00009_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00009_001.jpg -------------------------------------------------------------------------------- /src/00011_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00011_001.jpg -------------------------------------------------------------------------------- /src/00012_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00012_001.jpg -------------------------------------------------------------------------------- /src/00013_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00013_001.jpg -------------------------------------------------------------------------------- /src/00014_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00014_001.jpg -------------------------------------------------------------------------------- /src/00015_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00015_001.jpg -------------------------------------------------------------------------------- /src/00016_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00016_001.jpg -------------------------------------------------------------------------------- /src/00017_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00017_001.jpg -------------------------------------------------------------------------------- /src/00018_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00018_001.jpg -------------------------------------------------------------------------------- /src/00019_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00019_001.jpg -------------------------------------------------------------------------------- /src/00020_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00020_001.jpg -------------------------------------------------------------------------------- /src/00021_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00021_001.jpg -------------------------------------------------------------------------------- /src/00022_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00022_001.jpg -------------------------------------------------------------------------------- /src/00023_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00023_001.jpg -------------------------------------------------------------------------------- /src/00024_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00024_001.jpg -------------------------------------------------------------------------------- /src/00025_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00025_001.jpg -------------------------------------------------------------------------------- /src/00026_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00026_001.jpg -------------------------------------------------------------------------------- /src/00027_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00027_001.jpg -------------------------------------------------------------------------------- /src/00028_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00028_001.jpg -------------------------------------------------------------------------------- /src/00029_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00029_001.jpg -------------------------------------------------------------------------------- /src/00030_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00030_001.jpg -------------------------------------------------------------------------------- /src/00031_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00031_001.jpg -------------------------------------------------------------------------------- /src/00032_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00032_001.jpg -------------------------------------------------------------------------------- /src/00033_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00033_001.jpg -------------------------------------------------------------------------------- /src/00034_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00034_001.jpg -------------------------------------------------------------------------------- /src/00035_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00035_001.jpg -------------------------------------------------------------------------------- /src/00036_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00036_001.jpg -------------------------------------------------------------------------------- /src/00037_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00037_001.jpg -------------------------------------------------------------------------------- /src/00038_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00038_001.jpg -------------------------------------------------------------------------------- /src/00039_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00039_001.jpg -------------------------------------------------------------------------------- /src/00040_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00040_001.jpg -------------------------------------------------------------------------------- /src/00041_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00041_001.jpg -------------------------------------------------------------------------------- /src/00042_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00042_001.jpg -------------------------------------------------------------------------------- /src/00043_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00043_001.jpg -------------------------------------------------------------------------------- /src/00044_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00044_001.jpg -------------------------------------------------------------------------------- /src/00045_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00045_001.jpg -------------------------------------------------------------------------------- /src/00046_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00046_001.jpg -------------------------------------------------------------------------------- /src/00047_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00047_001.jpg -------------------------------------------------------------------------------- /src/00048_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00048_001.jpg -------------------------------------------------------------------------------- /src/00049_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00049_001.jpg -------------------------------------------------------------------------------- /src/00050_001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/src/00050_001.jpg -------------------------------------------------------------------------------- /train.configs/Model_PAF.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/train.configs/Model_PAF.pkl -------------------------------------------------------------------------------- /train.configs/keypoints_sim.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/train.configs/keypoints_sim.npy -------------------------------------------------------------------------------- /train.configs/param_whitening.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/train.configs/param_whitening.pkl -------------------------------------------------------------------------------- /train.configs/pncc_code.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/train.configs/pncc_code.npy -------------------------------------------------------------------------------- /train.configs/u_exp.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/train.configs/u_exp.npy -------------------------------------------------------------------------------- /train.configs/u_shp.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/train.configs/u_shp.npy -------------------------------------------------------------------------------- /train.configs/w_exp_sim.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/train.configs/w_exp_sim.npy -------------------------------------------------------------------------------- /train.configs/w_shp_sim.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/train.configs/w_shp_sim.npy -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/cv_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/cv_plot.py -------------------------------------------------------------------------------- /utils/cython/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/cython/mesh_core.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/cython/mesh_core.cpp -------------------------------------------------------------------------------- /utils/cython/mesh_core.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/cython/mesh_core.h -------------------------------------------------------------------------------- /utils/cython/mesh_core_cython.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/cython/mesh_core_cython.cpp -------------------------------------------------------------------------------- /utils/cython/mesh_core_cython.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/cython/mesh_core_cython.pyx -------------------------------------------------------------------------------- /utils/cython/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/cython/readme.md -------------------------------------------------------------------------------- /utils/cython/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/cython/setup.py -------------------------------------------------------------------------------- /utils/ddfa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/ddfa.py -------------------------------------------------------------------------------- /utils/estimate_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/estimate_pose.py -------------------------------------------------------------------------------- /utils/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/inference.py -------------------------------------------------------------------------------- /utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/io.py -------------------------------------------------------------------------------- /utils/lighting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/lighting.py -------------------------------------------------------------------------------- /utils/paf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/paf.py -------------------------------------------------------------------------------- /utils/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/params.py -------------------------------------------------------------------------------- /utils/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jhh37/wearmask3d/HEAD/utils/render.py --------------------------------------------------------------------------------