blazeface trained on pascal_voc person.
The training and inference is about 10x faster than SSD.
The BlazeFace model is based on https://github.com/tkat0/PyTorch_BlazeFace, and localaization and detection layers are added from ssd.pytorch.
Also the nms functions are from ssd.pytorch as well.
Thank you!
Pytorch > 1.0
opencv
scikit-learn
Run:
python train_with_blazeface.py
See Dataset_test_with_BlazeFace.ipynb for specifics.
Dataset_test_with_BlazeFace128-VOC-allclasses.ipynb runs all VOC classes, but doesn't get good accuracy at this point.
See inference.ipynb to run inference.
It takes less than 30ms on K80 to run single image (including nms).