├── .gitignore ├── 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 └── wider_face.py ├── detect.py ├── detect_dir.py ├── genwts.py ├── layers ├── __init__.py ├── functions │ └── prior_box.py └── modules │ ├── __init__.py │ └── multibox_loss.py ├── models ├── __init__.py ├── net.py └── retinaface.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 /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.MIT: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/LICENSE.MIT -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/README.md -------------------------------------------------------------------------------- /convert_to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/convert_to_onnx.py -------------------------------------------------------------------------------- /curve/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/curve/1.jpg -------------------------------------------------------------------------------- /curve/FDDB.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/curve/FDDB.png -------------------------------------------------------------------------------- /curve/Widerface.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/curve/Widerface.jpg -------------------------------------------------------------------------------- /curve/test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/curve/test.jpg -------------------------------------------------------------------------------- /data/FDDB/img_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/data/FDDB/img_list.txt -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/data/config.py -------------------------------------------------------------------------------- /data/data_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/data/data_augment.py -------------------------------------------------------------------------------- /data/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/data/wider_face.py -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/detect.py -------------------------------------------------------------------------------- /detect_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/detect_dir.py -------------------------------------------------------------------------------- /genwts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/genwts.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/functions/prior_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/layers/functions/prior_box.py -------------------------------------------------------------------------------- /layers/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/layers/modules/__init__.py -------------------------------------------------------------------------------- /layers/modules/multibox_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/layers/modules/multibox_loss.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/models/net.py -------------------------------------------------------------------------------- /models/retinaface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/models/retinaface.py -------------------------------------------------------------------------------- /test_fddb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/test_fddb.py -------------------------------------------------------------------------------- /test_widerface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/test_widerface.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/utils/box_utils.py -------------------------------------------------------------------------------- /utils/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/nms/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/utils/nms/py_cpu_nms.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/utils/timer.py -------------------------------------------------------------------------------- /widerface_evaluate/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/widerface_evaluate/README.md -------------------------------------------------------------------------------- /widerface_evaluate/box_overlaps.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/widerface_evaluate/box_overlaps.pyx -------------------------------------------------------------------------------- /widerface_evaluate/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/widerface_evaluate/evaluation.py -------------------------------------------------------------------------------- /widerface_evaluate/ground_truth/wider_easy_val.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/widerface_evaluate/ground_truth/wider_easy_val.mat -------------------------------------------------------------------------------- /widerface_evaluate/ground_truth/wider_face_val.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/widerface_evaluate/ground_truth/wider_face_val.mat -------------------------------------------------------------------------------- /widerface_evaluate/ground_truth/wider_hard_val.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/widerface_evaluate/ground_truth/wider_hard_val.mat -------------------------------------------------------------------------------- /widerface_evaluate/ground_truth/wider_medium_val.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/widerface_evaluate/ground_truth/wider_medium_val.mat -------------------------------------------------------------------------------- /widerface_evaluate/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wang-xinyu/Pytorch_Retinaface/HEAD/widerface_evaluate/setup.py --------------------------------------------------------------------------------