├── OptimalQualityFunctions.png ├── README.md ├── Table_Benchmarks.png ├── Table_QualityAwareness.png ├── _data └── single_images │ └── fix_filenames.py ├── _models └── mtcnn-model │ ├── det1-0001.params │ ├── det1-symbol.json │ ├── det1.caffemodel │ ├── det1.prototxt │ ├── det2-0001.params │ ├── det2-symbol.json │ ├── det2.caffemodel │ ├── det2.prototxt │ ├── det3-0001.params │ ├── det3-symbol.json │ ├── det3.caffemodel │ ├── det3.prototxt │ ├── det4-0001.params │ ├── det4-symbol.json │ ├── det4.caffemodel │ └── det4.prototxt ├── align_images.py ├── datasets ├── __init__.py ├── paired.py └── single.py ├── embed_images.py ├── environment.yml ├── evaluation ├── __init__.py ├── metrics.py └── pairs_parsing.py ├── experiments ├── __init__.py ├── ijb_benchmarks.py ├── memory_analysis.py ├── plots │ ├── __init__.py │ ├── db_utils.py │ └── training_visualization.py ├── robustness_analysis.py ├── runtime_analysis.py └── single_image_benchmarks.py ├── launch_experiments.py ├── magface ├── __init__.py ├── inference │ ├── __init__.py │ ├── gen_feat.py │ ├── generate_embeddings.py │ ├── network_inf.py │ └── toy_imgs │ │ ├── 0.jpg │ │ ├── 1.jpg │ │ ├── 2.jpg │ │ ├── 3.jpg │ │ ├── 4.jpg │ │ ├── 5.jpg │ │ ├── 6.jpg │ │ ├── 7.jpg │ │ ├── 8.jpg │ │ ├── 9.jpg │ │ ├── feat.list │ │ └── img.list ├── models │ ├── __init__.py │ └── iresnet.py └── utils │ ├── __init__.py │ └── utils.py ├── preprocessing ├── __init__.py ├── align.py ├── embed.py ├── insightface │ ├── __init__.py │ ├── license.txt │ └── src │ │ ├── __init__.py │ │ ├── face_preprocess.py │ │ ├── helper.py │ │ └── mtcnn_detector.py └── magface │ ├── __init__.py │ ├── gen_feat.py │ ├── iresnet.py │ ├── network_inf.py │ └── utils.py ├── similarity ├── __init__.py ├── base.py ├── cosine.py └── qmagface.py ├── utils ├── __init__.py ├── files.py ├── pairs_files.py └── setup.py └── visualization with explanation.png /OptimalQualityFunctions.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/OptimalQualityFunctions.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/README.md -------------------------------------------------------------------------------- /Table_Benchmarks.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/Table_Benchmarks.png -------------------------------------------------------------------------------- /Table_QualityAwareness.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/Table_QualityAwareness.png -------------------------------------------------------------------------------- /_data/single_images/fix_filenames.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_data/single_images/fix_filenames.py -------------------------------------------------------------------------------- /_models/mtcnn-model/det1-0001.params: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det1-0001.params -------------------------------------------------------------------------------- /_models/mtcnn-model/det1-symbol.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det1-symbol.json -------------------------------------------------------------------------------- /_models/mtcnn-model/det1.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det1.caffemodel -------------------------------------------------------------------------------- /_models/mtcnn-model/det1.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det1.prototxt -------------------------------------------------------------------------------- /_models/mtcnn-model/det2-0001.params: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det2-0001.params -------------------------------------------------------------------------------- /_models/mtcnn-model/det2-symbol.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det2-symbol.json -------------------------------------------------------------------------------- /_models/mtcnn-model/det2.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det2.caffemodel -------------------------------------------------------------------------------- /_models/mtcnn-model/det2.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det2.prototxt -------------------------------------------------------------------------------- /_models/mtcnn-model/det3-0001.params: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det3-0001.params -------------------------------------------------------------------------------- /_models/mtcnn-model/det3-symbol.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det3-symbol.json -------------------------------------------------------------------------------- /_models/mtcnn-model/det3.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det3.caffemodel -------------------------------------------------------------------------------- /_models/mtcnn-model/det3.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det3.prototxt -------------------------------------------------------------------------------- /_models/mtcnn-model/det4-0001.params: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det4-0001.params -------------------------------------------------------------------------------- /_models/mtcnn-model/det4-symbol.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det4-symbol.json -------------------------------------------------------------------------------- /_models/mtcnn-model/det4.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det4.caffemodel -------------------------------------------------------------------------------- /_models/mtcnn-model/det4.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/_models/mtcnn-model/det4.prototxt -------------------------------------------------------------------------------- /align_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/align_images.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/paired.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/datasets/paired.py -------------------------------------------------------------------------------- /datasets/single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/datasets/single.py -------------------------------------------------------------------------------- /embed_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/embed_images.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/environment.yml -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/evaluation/metrics.py -------------------------------------------------------------------------------- /evaluation/pairs_parsing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/evaluation/pairs_parsing.py -------------------------------------------------------------------------------- /experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/ijb_benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/experiments/ijb_benchmarks.py -------------------------------------------------------------------------------- /experiments/memory_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/experiments/memory_analysis.py -------------------------------------------------------------------------------- /experiments/plots/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /experiments/plots/db_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/experiments/plots/db_utils.py -------------------------------------------------------------------------------- /experiments/plots/training_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/experiments/plots/training_visualization.py -------------------------------------------------------------------------------- /experiments/robustness_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/experiments/robustness_analysis.py -------------------------------------------------------------------------------- /experiments/runtime_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/experiments/runtime_analysis.py -------------------------------------------------------------------------------- /experiments/single_image_benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/experiments/single_image_benchmarks.py -------------------------------------------------------------------------------- /launch_experiments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/launch_experiments.py -------------------------------------------------------------------------------- /magface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magface/inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magface/inference/gen_feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/gen_feat.py -------------------------------------------------------------------------------- /magface/inference/generate_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/generate_embeddings.py -------------------------------------------------------------------------------- /magface/inference/network_inf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/network_inf.py -------------------------------------------------------------------------------- /magface/inference/toy_imgs/0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/0.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/1.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/2.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/3.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/4.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/5.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/6.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/6.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/7.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/7.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/8.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/8.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/9.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/9.jpg -------------------------------------------------------------------------------- /magface/inference/toy_imgs/feat.list: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magface/inference/toy_imgs/img.list: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/inference/toy_imgs/img.list -------------------------------------------------------------------------------- /magface/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magface/models/iresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/models/iresnet.py -------------------------------------------------------------------------------- /magface/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /magface/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/magface/utils/utils.py -------------------------------------------------------------------------------- /preprocessing/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /preprocessing/align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/align.py -------------------------------------------------------------------------------- /preprocessing/embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/embed.py -------------------------------------------------------------------------------- /preprocessing/insightface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /preprocessing/insightface/license.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/insightface/license.txt -------------------------------------------------------------------------------- /preprocessing/insightface/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /preprocessing/insightface/src/face_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/insightface/src/face_preprocess.py -------------------------------------------------------------------------------- /preprocessing/insightface/src/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/insightface/src/helper.py -------------------------------------------------------------------------------- /preprocessing/insightface/src/mtcnn_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/insightface/src/mtcnn_detector.py -------------------------------------------------------------------------------- /preprocessing/magface/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /preprocessing/magface/gen_feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/magface/gen_feat.py -------------------------------------------------------------------------------- /preprocessing/magface/iresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/magface/iresnet.py -------------------------------------------------------------------------------- /preprocessing/magface/network_inf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/magface/network_inf.py -------------------------------------------------------------------------------- /preprocessing/magface/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/preprocessing/magface/utils.py -------------------------------------------------------------------------------- /similarity/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /similarity/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/similarity/base.py -------------------------------------------------------------------------------- /similarity/cosine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/similarity/cosine.py -------------------------------------------------------------------------------- /similarity/qmagface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/similarity/qmagface.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/utils/files.py -------------------------------------------------------------------------------- /utils/pairs_files.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/utils/pairs_files.py -------------------------------------------------------------------------------- /utils/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/utils/setup.py -------------------------------------------------------------------------------- /visualization with explanation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pterhoer/QMagFace/HEAD/visualization with explanation.png --------------------------------------------------------------------------------