├── .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 ├── images ├── example.png ├── office1.jpg ├── office2.jpg ├── office3.jpg ├── office4.jpg └── office5.jpg ├── src ├── __init__.py ├── box_utils.py ├── detector.py ├── first_stage.py ├── get_nets.py ├── visualization_utils.py └── weights │ ├── onet.npy │ ├── pnet.npy │ └── rnet.npy ├── test_on_images.ipynb └── try_mtcnn_step_by_step.ipynb /.gitignore: -------------------------------------------------------------------------------- 1 | .ipynb_checkpoints 2 | __pycache__ 3 | 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /caffe_models/det1.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/caffe_models/det1.caffemodel -------------------------------------------------------------------------------- /caffe_models/det1.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/caffe_models/det1.prototxt -------------------------------------------------------------------------------- /caffe_models/det2.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/caffe_models/det2.caffemodel -------------------------------------------------------------------------------- /caffe_models/det2.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/caffe_models/det2.prototxt -------------------------------------------------------------------------------- /caffe_models/det3.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/caffe_models/det3.caffemodel -------------------------------------------------------------------------------- /caffe_models/det3.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/caffe_models/det3.prototxt -------------------------------------------------------------------------------- /caffe_models/det4.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/caffe_models/det4.caffemodel -------------------------------------------------------------------------------- /caffe_models/det4.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/caffe_models/det4.prototxt -------------------------------------------------------------------------------- /extract_weights_from_caffe_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/extract_weights_from_caffe_models.py -------------------------------------------------------------------------------- /images/example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/images/example.png -------------------------------------------------------------------------------- /images/office1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/images/office1.jpg -------------------------------------------------------------------------------- /images/office2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/images/office2.jpg -------------------------------------------------------------------------------- /images/office3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/images/office3.jpg -------------------------------------------------------------------------------- /images/office4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/images/office4.jpg -------------------------------------------------------------------------------- /images/office5.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/images/office5.jpg -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/src/__init__.py -------------------------------------------------------------------------------- /src/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/src/box_utils.py -------------------------------------------------------------------------------- /src/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/src/detector.py -------------------------------------------------------------------------------- /src/first_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/src/first_stage.py -------------------------------------------------------------------------------- /src/get_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/src/get_nets.py -------------------------------------------------------------------------------- /src/visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/src/visualization_utils.py -------------------------------------------------------------------------------- /src/weights/onet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/src/weights/onet.npy -------------------------------------------------------------------------------- /src/weights/pnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/src/weights/pnet.npy -------------------------------------------------------------------------------- /src/weights/rnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/src/weights/rnet.npy -------------------------------------------------------------------------------- /test_on_images.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/test_on_images.ipynb -------------------------------------------------------------------------------- /try_mtcnn_step_by_step.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TropComplique/mtcnn-pytorch/HEAD/try_mtcnn_step_by_step.ipynb --------------------------------------------------------------------------------