face-detection-ssd-mobilenet

face-detection-ssd-mobilenet-tensorflow

Please install Tensorflow Object Detection API first: https://github.com/tensorflow/models/tree/master/research/object_detection remember to export PYTHONPATH in .profile like: export PYTHONPATH=$PYTHONPATH:/Users/ls/TF_Pro/models/research:/Users/ls/TF_Pro/models/research/slim

Prepare

python 1_download_data.py

python3 2_data_to_pascal_xml.py

python 3_xml_to_csv.py

python 4_generate_tfrecord.py --images_path=data/tf_wider_train/images --csv_input=data/tf_wider_train/train.csv --output_path=data/train.record

python 4_generate_tfrecord.py --images_path=data/tf_wider_val/images --csv_input=data/tf_wider_val/val.csv --output_path=data/val.record

Modify

Read the comments and modify the config information in ssd_mobilenet_v1_face.config

Train

python /Users/ls/TF_Pro/models/research/object_detection/train.py --logtostderr --pipeline_config_path=ssd_mobilenet_v1_face.config --train_dir=checkpoints_dir

Export Model

python /Users/ls/TF_Pro/models/research/object_detection/export_inference_graph.py --input_type image_tensor --pipeline_config_path ssd_mobilenet_v1_face.config --trained_checkpoint_prefix checkpoints_dir/model.ckpt-200 --output_directory output_model/

Please modify the name of trained_checkpoint_prefix: checkpoints_dir/model.ckpt-xxx, xxx is the num_step in config file

Eval

python /Users/ls/TF_Pro/models/research/object_detection/eval.py --logtostderr --pipeline_config_path=ssd_mobilenet_v1_face.config --checkpoint_dir=checkpoints_dir --eval_dir=eval

Run

python detect_face.py