├── README.md ├── configs └── detection │ └── rtmdet_tiny_8xb32-300e_coco.py ├── datasets_utils └── build │ ├── __pycache__ │ └── detect_utils.cpython-310.pyc │ ├── detect_utils.py │ └── seg_utils.py ├── detect.py ├── docs └── pictures │ └── instantbooth.png ├── face_alignment ├── __pycache__ │ ├── align.cpython-310.pyc │ └── mtcnn.cpython-310.pyc ├── align.py ├── mtcnn.py └── mtcnn_pytorch │ ├── caffe_models │ ├── det1.caffemodel │ ├── det1.prototxt │ ├── det2.caffemodel │ ├── det2.prototxt │ ├── det3.caffemodel │ ├── det3.prototxt │ ├── det4.caffemodel │ └── det4.prototxt │ └── src │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-310.pyc │ ├── align_trans.cpython-310.pyc │ ├── box_utils.cpython-310.pyc │ ├── detector.cpython-310.pyc │ ├── first_stage.cpython-310.pyc │ ├── get_nets.cpython-310.pyc │ ├── matlab_cp2tform.cpython-310.pyc │ └── visualization_utils.cpython-310.pyc │ ├── align_trans.py │ ├── box_utils.py │ ├── detector.py │ ├── first_stage.py │ ├── get_nets.py │ ├── matlab_cp2tform.py │ ├── visualization_utils.py │ └── weights │ ├── onet.npy │ ├── pnet.npy │ └── rnet.npy ├── images ├── 2people.jpg ├── OIP-C.jpg ├── R.jpg ├── image1.jpg └── woman.jpg ├── test.py └── weights └── detection └── rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/README.md -------------------------------------------------------------------------------- /configs/detection/rtmdet_tiny_8xb32-300e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/configs/detection/rtmdet_tiny_8xb32-300e_coco.py -------------------------------------------------------------------------------- /datasets_utils/build/__pycache__/detect_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/datasets_utils/build/__pycache__/detect_utils.cpython-310.pyc -------------------------------------------------------------------------------- /datasets_utils/build/detect_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/datasets_utils/build/detect_utils.py -------------------------------------------------------------------------------- /datasets_utils/build/seg_utils.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/detect.py -------------------------------------------------------------------------------- /docs/pictures/instantbooth.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/docs/pictures/instantbooth.png -------------------------------------------------------------------------------- /face_alignment/__pycache__/align.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/__pycache__/align.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/__pycache__/mtcnn.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/__pycache__/mtcnn.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/align.py -------------------------------------------------------------------------------- /face_alignment/mtcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn.py -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/caffe_models/det1.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/caffe_models/det1.caffemodel -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/caffe_models/det1.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/caffe_models/det1.prototxt -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/caffe_models/det2.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/caffe_models/det2.caffemodel -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/caffe_models/det2.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/caffe_models/det2.prototxt -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/caffe_models/det3.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/caffe_models/det3.caffemodel -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/caffe_models/det3.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/caffe_models/det3.prototxt -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/caffe_models/det4.caffemodel: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/caffe_models/det4.caffemodel -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/caffe_models/det4.prototxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/caffe_models/det4.prototxt -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/__init__.py -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/__pycache__/__init__.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/__pycache__/__init__.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/__pycache__/align_trans.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/__pycache__/align_trans.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/__pycache__/box_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/__pycache__/box_utils.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/__pycache__/detector.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/__pycache__/detector.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/__pycache__/first_stage.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/__pycache__/first_stage.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/__pycache__/get_nets.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/__pycache__/get_nets.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/__pycache__/matlab_cp2tform.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/__pycache__/matlab_cp2tform.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/__pycache__/visualization_utils.cpython-310.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/__pycache__/visualization_utils.cpython-310.pyc -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/align_trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/align_trans.py -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/box_utils.py -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/detector.py -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/first_stage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/first_stage.py -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/get_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/get_nets.py -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/matlab_cp2tform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/matlab_cp2tform.py -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/visualization_utils.py -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/weights/onet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/weights/onet.npy -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/weights/pnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/weights/pnet.npy -------------------------------------------------------------------------------- /face_alignment/mtcnn_pytorch/src/weights/rnet.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/face_alignment/mtcnn_pytorch/src/weights/rnet.npy -------------------------------------------------------------------------------- /images/2people.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/images/2people.jpg -------------------------------------------------------------------------------- /images/OIP-C.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/images/OIP-C.jpg -------------------------------------------------------------------------------- /images/R.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/images/R.jpg -------------------------------------------------------------------------------- /images/image1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/images/image1.jpg -------------------------------------------------------------------------------- /images/woman.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/images/woman.jpg -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/test.py -------------------------------------------------------------------------------- /weights/detection/rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zcai0612/InstantBooth/HEAD/weights/detection/rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth --------------------------------------------------------------------------------