├── .gitignore ├── LICENSE ├── README.md ├── datasets.py ├── evaluation ├── __init__.py ├── compute_roc.py ├── megaface.py └── verify.py ├── experiments ├── ms1m │ ├── densenet121-arc-bs64-sz224-ep14 │ │ ├── config.yaml │ │ ├── evaluate.sh │ │ ├── extract.sh │ │ ├── resume.sh │ │ └── train.sh │ └── densenet121-bs64-sz224-ep14 │ │ ├── config.yaml │ │ ├── evaluate.sh │ │ ├── extract.sh │ │ ├── resume.sh │ │ └── train.sh └── webface │ └── res50-bs64-sz224-ep35 │ ├── config.yaml │ ├── evaluate.sh │ ├── extract.sh │ ├── resume.sh │ └── train.sh ├── main.py ├── models ├── __init__.py ├── backbones │ ├── __init__.py │ ├── densenet.py │ ├── inception.py │ ├── ir.py │ ├── lookup.py │ ├── nasnet.py │ ├── resnet.py │ └── vgg.py ├── ext_layers │ ├── __init__.py │ └── arc_fc_layer.py └── multitask.py ├── scripts └── data_stats.ipynb ├── tools ├── __init__.py ├── convert_data.py ├── convert_data_para.py ├── ensemble.py ├── param_stat.py ├── split_id_even.py ├── split_id_ratio.py └── visualize.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/README.md -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/datasets.py -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/evaluation/__init__.py -------------------------------------------------------------------------------- /evaluation/compute_roc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/evaluation/compute_roc.py -------------------------------------------------------------------------------- /evaluation/megaface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/evaluation/megaface.py -------------------------------------------------------------------------------- /evaluation/verify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/evaluation/verify.py -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-arc-bs64-sz224-ep14/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-arc-bs64-sz224-ep14/config.yaml -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-arc-bs64-sz224-ep14/evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-arc-bs64-sz224-ep14/evaluate.sh -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-arc-bs64-sz224-ep14/extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-arc-bs64-sz224-ep14/extract.sh -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-arc-bs64-sz224-ep14/resume.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-arc-bs64-sz224-ep14/resume.sh -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-arc-bs64-sz224-ep14/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-arc-bs64-sz224-ep14/train.sh -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-bs64-sz224-ep14/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-bs64-sz224-ep14/config.yaml -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-bs64-sz224-ep14/evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-bs64-sz224-ep14/evaluate.sh -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-bs64-sz224-ep14/extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-bs64-sz224-ep14/extract.sh -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-bs64-sz224-ep14/resume.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-bs64-sz224-ep14/resume.sh -------------------------------------------------------------------------------- /experiments/ms1m/densenet121-bs64-sz224-ep14/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/ms1m/densenet121-bs64-sz224-ep14/train.sh -------------------------------------------------------------------------------- /experiments/webface/res50-bs64-sz224-ep35/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/webface/res50-bs64-sz224-ep35/config.yaml -------------------------------------------------------------------------------- /experiments/webface/res50-bs64-sz224-ep35/evaluate.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/webface/res50-bs64-sz224-ep35/evaluate.sh -------------------------------------------------------------------------------- /experiments/webface/res50-bs64-sz224-ep35/extract.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/webface/res50-bs64-sz224-ep35/extract.sh -------------------------------------------------------------------------------- /experiments/webface/res50-bs64-sz224-ep35/resume.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/webface/res50-bs64-sz224-ep35/resume.sh -------------------------------------------------------------------------------- /experiments/webface/res50-bs64-sz224-ep35/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/experiments/webface/res50-bs64-sz224-ep35/train.sh -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | from .multitask import * 2 | -------------------------------------------------------------------------------- /models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/backbones/__init__.py -------------------------------------------------------------------------------- /models/backbones/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/backbones/densenet.py -------------------------------------------------------------------------------- /models/backbones/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/backbones/inception.py -------------------------------------------------------------------------------- /models/backbones/ir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/backbones/ir.py -------------------------------------------------------------------------------- /models/backbones/lookup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/backbones/lookup.py -------------------------------------------------------------------------------- /models/backbones/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/backbones/nasnet.py -------------------------------------------------------------------------------- /models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/backbones/resnet.py -------------------------------------------------------------------------------- /models/backbones/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/backbones/vgg.py -------------------------------------------------------------------------------- /models/ext_layers/__init__.py: -------------------------------------------------------------------------------- 1 | from .arc_fc_layer import ArcFullyConnected 2 | -------------------------------------------------------------------------------- /models/ext_layers/arc_fc_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/ext_layers/arc_fc_layer.py -------------------------------------------------------------------------------- /models/multitask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/models/multitask.py -------------------------------------------------------------------------------- /scripts/data_stats.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/scripts/data_stats.ipynb -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/convert_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/tools/convert_data.py -------------------------------------------------------------------------------- /tools/convert_data_para.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/tools/convert_data_para.py -------------------------------------------------------------------------------- /tools/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/tools/ensemble.py -------------------------------------------------------------------------------- /tools/param_stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/tools/param_stat.py -------------------------------------------------------------------------------- /tools/split_id_even.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/tools/split_id_even.py -------------------------------------------------------------------------------- /tools/split_id_ratio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/tools/split_id_ratio.py -------------------------------------------------------------------------------- /tools/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/tools/visualize.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaohangZhan/face_recognition_framework/HEAD/utils.py --------------------------------------------------------------------------------