This crate implements MTCNN facial and facial feature detection in the rust programming language. It is based on the python demo of this neural network. Nvidia's Tensorrt rust wrappings are provided by the repo of Mason Stallmo. Currently the Cargo.toml file links to this local version, since this library is extended to fit the needs of this crate. Neural network files are taken from the python demo and converted to an optimized tensorrt dump (which matters for the hardware it was ran on). Current development state:
- Pnet functionality is written, needs to be integrated
- Rnet not implemented yet
- Onet not implemented yet
Build create_engines for windows: nvcc -I ../common create_engines.cpp -o create_engines -lnvinfer -lnvparsers -lnvinfer_plugin