├── acceleration ├── CMakeLists.txt ├── cuda_kernels.cu ├── cuda_kernels.hpp └── cuda_postprocess.cpp ├── face_modules ├── .gitignore ├── ReadMe.txt ├── __init__.py ├── infer_demo.py ├── model.py ├── mtcnn.py ├── mtcnn_pytorch │ ├── .gitignore │ ├── LICENSE │ ├── README.md │ ├── caffe_models │ │ ├── det1.caffemodel │ │ ├── det1.prototxt │ │ ├── det2.caffemodel │ │ ├── det2.prototxt │ │ ├── det3.caffemodel │ │ ├── det3.prototxt │ │ ├── det4.caffemodel │ │ └── det4.prototxt │ ├── extract_weights_from_caffe_models.py │ ├── get_aligned_face_from_mtcnn.ipynb │ ├── images │ │ ├── example.png │ │ ├── jf.jpg │ │ ├── office1.jpg │ │ ├── office2.jpg │ │ ├── office3.jpg │ │ ├── office4.jpg │ │ └── office5.jpg │ ├── refine_faces.ipynb │ ├── src │ │ ├── __init__.py │ │ ├── align_trans.py │ │ ├── box_utils.py │ │ ├── detector.py │ │ ├── first_stage.py │ │ ├── get_nets.py │ │ ├── matlab_cp2tform.py │ │ ├── visualization_utils.py │ │ └── weights │ │ │ ├── onet.npy │ │ │ ├── pnet.npy │ │ │ └── rnet.npy │ ├── test_on_images.ipynb │ └── try_mtcnn_step_by_step.ipynb └── preprocess_images.py ├── inference_demo.py ├── mtcnn_pytorch ├── network ├── AADLayer.py ├── AEI_Net.py ├── HEAR_Net.py ├── MultiscaleDiscriminator.py └── __init__.py ├── online_preview.py ├── tmp_script ├── __init__.py ├── check_arcface_feature_map.py └── test_structure_AEI.py ├── train_AEI.py ├── train_HEAR.py └── utils ├── Dataset.py ├── download_vggface_dataset.py └── split_hearnet_data.py /acceleration/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/acceleration/CMakeLists.txt -------------------------------------------------------------------------------- /acceleration/cuda_kernels.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/acceleration/cuda_kernels.cu -------------------------------------------------------------------------------- /acceleration/cuda_kernels.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/acceleration/cuda_kernels.hpp -------------------------------------------------------------------------------- /acceleration/cuda_postprocess.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/acceleration/cuda_postprocess.cpp -------------------------------------------------------------------------------- /face_modules/.gitignore: -------------------------------------------------------------------------------- 1 | model_ir_se50.pth 2 | libnvjpeg.cpython-36m-x86_64-linux-gnu.so 3 | -------------------------------------------------------------------------------- /face_modules/ReadMe.txt: -------------------------------------------------------------------------------- 1 | https://github.com/TreB1eN/InsightFace_Pytorch 2 | -------------------------------------------------------------------------------- /face_modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /face_modules/infer_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/infer_demo.py -------------------------------------------------------------------------------- /face_modules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/model.py -------------------------------------------------------------------------------- /face_modules/mtcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/.gitignore: -------------------------------------------------------------------------------- 1 | .ipynb_checkpoints 2 | __pycache__ 3 | 4 | -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/LICENSE -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/README.md -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/caffe_models/det1.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/caffe_models/det1.caffemodel -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/caffe_models/det1.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/caffe_models/det1.prototxt -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/caffe_models/det2.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/caffe_models/det2.caffemodel -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/caffe_models/det2.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/caffe_models/det2.prototxt -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/caffe_models/det3.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/caffe_models/det3.caffemodel -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/caffe_models/det3.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/caffe_models/det3.prototxt -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/caffe_models/det4.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/caffe_models/det4.caffemodel -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/caffe_models/det4.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/caffe_models/det4.prototxt -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/extract_weights_from_caffe_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/extract_weights_from_caffe_models.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/get_aligned_face_from_mtcnn.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/get_aligned_face_from_mtcnn.ipynb -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/images/example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/images/example.png -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/images/jf.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/images/jf.jpg -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/images/office1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/images/office1.jpg -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/images/office2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/images/office2.jpg -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/images/office3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/images/office3.jpg -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/images/office4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/images/office4.jpg -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/images/office5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/images/office5.jpg -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/refine_faces.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/refine_faces.ipynb -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/__init__.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/align_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/align_trans.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/box_utils.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/detector.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/first_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/first_stage.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/get_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/get_nets.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/matlab_cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/matlab_cp2tform.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/visualization_utils.py -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/weights/onet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/weights/onet.npy -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/weights/pnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/weights/pnet.npy -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/src/weights/rnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/src/weights/rnet.npy -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/test_on_images.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/test_on_images.ipynb -------------------------------------------------------------------------------- /face_modules/mtcnn_pytorch/try_mtcnn_step_by_step.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/mtcnn_pytorch/try_mtcnn_step_by_step.ipynb -------------------------------------------------------------------------------- /face_modules/preprocess_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/face_modules/preprocess_images.py -------------------------------------------------------------------------------- /inference_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/inference_demo.py -------------------------------------------------------------------------------- /mtcnn_pytorch: -------------------------------------------------------------------------------- 1 | ./face_modules/mtcnn_pytorch/ -------------------------------------------------------------------------------- /network/AADLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/network/AADLayer.py -------------------------------------------------------------------------------- /network/AEI_Net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/network/AEI_Net.py -------------------------------------------------------------------------------- /network/HEAR_Net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/network/HEAR_Net.py -------------------------------------------------------------------------------- /network/MultiscaleDiscriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/network/MultiscaleDiscriminator.py -------------------------------------------------------------------------------- /network/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /online_preview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/online_preview.py -------------------------------------------------------------------------------- /tmp_script/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tmp_script/check_arcface_feature_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/tmp_script/check_arcface_feature_map.py -------------------------------------------------------------------------------- /tmp_script/test_structure_AEI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/tmp_script/test_structure_AEI.py -------------------------------------------------------------------------------- /train_AEI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/train_AEI.py -------------------------------------------------------------------------------- /train_HEAR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/train_HEAR.py -------------------------------------------------------------------------------- /utils/Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/utils/Dataset.py -------------------------------------------------------------------------------- /utils/download_vggface_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/utils/download_vggface_dataset.py -------------------------------------------------------------------------------- /utils/split_hearnet_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taotaonice/FaceShifter/HEAD/utils/split_hearnet_data.py --------------------------------------------------------------------------------