The package attempts to count faces using face detection and motion detection. It will be used on vending machines to calculate product attention and get traffic at that location.

pip install -r requirements.txt
# Need to download models when using real-time pose estimation.
bash model_download.sh

Use face_counter.py -

python face_counter.py

Use motion_detection.py

python motion_detection.py

Use transparent.py

python transparent.py

Use object_detection.py

python object_detection.py --prototxt data/MobileNetSSD/MobileNetSSD_deploy.prototxt.txt --model data/MobileNetSSD/MobileNetSSD_deploy.caffemodel

Use calibrate_camera.py

python calibrate_camera.py

Use wide_angle_repair.py

python wide_angle_repair.py

Use pose_estimation.py

• MPI pretrained model

python pose_estimation.py --proto data/mpi/pose_deploy_linevec_faster_4_stages.prototxt  --model data/mpi/pose_iter_160000.caffemodel --dataset MPI

• Body_25 pretrained model

python pose_estimation.py --proto data/body_25/body_25_deploy.prototxt  --model data/body_25/pose_iter_584000.caffemodel

• COCO pretrained model

python pose_estimation.py --proto data/coco/deploy_coco.prototxt  --model data/coco/pose_iter_440000.caffemodel --dataset COCO

Overlay image enable (Boolean)

overlay_enable = True

Pixel blurring (Tuple)

blur_pixel = (4, 4)

Move detection thresh (Int) range: 1 - 255

move_detection_thresh = 64

Minimum size of moving filter (Int) range: 0 - width*height

move_min_size = 2500

Face detection times (Int) range: > 1

face_detection_times = 3 

Minimum pixel of face filter (Tuple)

face_min_pixel = (120, 120)

Face detection interval (Int) range: > 1

face_detection_interval = 1

Motion detection average adjustment (Float)

avg_adjustment = 0.2

Webcam frame width (Int)

width = 1280

Webcam frame height (Int)

height = 960

OpenCV data, from https://github.com/opencv/opencv

MobileNetSSD caffemodel, from https://github.com/PINTO0309/MobileNet-SSD-RealSense

Wide Angle Repair, from https://blog.csdn.net/donkey_1993/article/details/103909811

Calibrate Camera, from https://blog.csdn.net/Thomson617/article/details/103506391

Pose Estimation, from https://github.com/legolas123/cv-tricks.com