├── LICENSE.MIT ├── README.md ├── convert_to_onnx.py ├── curve ├── 1.jpg ├── FDDB.png ├── Widerface.jpg └── test.jpg ├── data ├── FDDB │ └── img_list.txt ├── __init__.py ├── config.py ├── data_augment.py ├── input │ └── huge.jpg ├── output │ ├── bbox_pred_acc.png │ └── test_5.jpg └── wider_face.py ├── detect.py ├── detect_merge.py ├── layers ├── __init__.py ├── functions │ └── prior_box.py └── modules │ ├── __init__.py │ └── multibox_loss.py ├── models ├── __init__.py ├── net.py └── retinaface.py ├── pose ├── __init__.py ├── datasets.py ├── detect_image.py ├── hopenet.py ├── test_alexnet.py ├── test_hopenet.py ├── test_on_video.py ├── test_on_video_dlib.py ├── test_on_video_dockerface.py ├── test_resnet50_regression.py ├── train_alexnet.py ├── train_hopenet.py ├── train_resnet50_regression.py └── utils.py ├── test_fddb.py ├── test_widerface.py ├── train.py ├── utils ├── __init__.py ├── box_utils.py ├── nms │ ├── __init__.py │ └── py_cpu_nms.py └── timer.py └── widerface_evaluate ├── README.md ├── box_overlaps.pyx ├── evaluation.py ├── ground_truth ├── wider_easy_val.mat ├── wider_face_val.mat ├── wider_hard_val.mat └── wider_medium_val.mat └── setup.py /LICENSE.MIT: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/LICENSE.MIT -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/README.md -------------------------------------------------------------------------------- /convert_to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/convert_to_onnx.py -------------------------------------------------------------------------------- /curve/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/curve/1.jpg -------------------------------------------------------------------------------- /curve/FDDB.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/curve/FDDB.png -------------------------------------------------------------------------------- /curve/Widerface.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/curve/Widerface.jpg -------------------------------------------------------------------------------- /curve/test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/curve/test.jpg -------------------------------------------------------------------------------- /data/FDDB/img_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/data/FDDB/img_list.txt -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/data/config.py -------------------------------------------------------------------------------- /data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/data/data_augment.py -------------------------------------------------------------------------------- /data/input/huge.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/data/input/huge.jpg -------------------------------------------------------------------------------- /data/output/bbox_pred_acc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/data/output/bbox_pred_acc.png -------------------------------------------------------------------------------- /data/output/test_5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/data/output/test_5.jpg -------------------------------------------------------------------------------- /data/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/data/wider_face.py -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/detect.py -------------------------------------------------------------------------------- /detect_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/detect_merge.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/functions/prior_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/layers/functions/prior_box.py -------------------------------------------------------------------------------- /layers/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/layers/modules/__init__.py -------------------------------------------------------------------------------- /layers/modules/multibox_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/layers/modules/multibox_loss.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/models/net.py -------------------------------------------------------------------------------- /models/retinaface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/models/retinaface.py -------------------------------------------------------------------------------- /pose/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pose/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/datasets.py -------------------------------------------------------------------------------- /pose/detect_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/detect_image.py -------------------------------------------------------------------------------- /pose/hopenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/hopenet.py -------------------------------------------------------------------------------- /pose/test_alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/test_alexnet.py -------------------------------------------------------------------------------- /pose/test_hopenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/test_hopenet.py -------------------------------------------------------------------------------- /pose/test_on_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/test_on_video.py -------------------------------------------------------------------------------- /pose/test_on_video_dlib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/test_on_video_dlib.py -------------------------------------------------------------------------------- /pose/test_on_video_dockerface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/test_on_video_dockerface.py -------------------------------------------------------------------------------- /pose/test_resnet50_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/test_resnet50_regression.py -------------------------------------------------------------------------------- /pose/train_alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/train_alexnet.py -------------------------------------------------------------------------------- /pose/train_hopenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/train_hopenet.py -------------------------------------------------------------------------------- /pose/train_resnet50_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/train_resnet50_regression.py -------------------------------------------------------------------------------- /pose/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/pose/utils.py -------------------------------------------------------------------------------- /test_fddb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/test_fddb.py -------------------------------------------------------------------------------- /test_widerface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/test_widerface.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/utils/box_utils.py -------------------------------------------------------------------------------- /utils/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/nms/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/utils/nms/py_cpu_nms.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/utils/timer.py -------------------------------------------------------------------------------- /widerface_evaluate/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/widerface_evaluate/README.md -------------------------------------------------------------------------------- /widerface_evaluate/box_overlaps.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/widerface_evaluate/box_overlaps.pyx -------------------------------------------------------------------------------- /widerface_evaluate/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/widerface_evaluate/evaluation.py -------------------------------------------------------------------------------- /widerface_evaluate/ground_truth/wider_easy_val.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/widerface_evaluate/ground_truth/wider_easy_val.mat -------------------------------------------------------------------------------- /widerface_evaluate/ground_truth/wider_face_val.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/widerface_evaluate/ground_truth/wider_face_val.mat -------------------------------------------------------------------------------- /widerface_evaluate/ground_truth/wider_hard_val.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/widerface_evaluate/ground_truth/wider_hard_val.mat -------------------------------------------------------------------------------- /widerface_evaluate/ground_truth/wider_medium_val.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/widerface_evaluate/ground_truth/wider_medium_val.mat -------------------------------------------------------------------------------- /widerface_evaluate/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gengyanlei/Pytorch_Retinaface/HEAD/widerface_evaluate/setup.py --------------------------------------------------------------------------------