chandrikadeb7 / Face-Mask-Detection

Face Mask Detection system based on computer vision and deep learning using OpenCV and Tensorflow/Keras
https://www.youtube.com/watch?v=AAkNyZlUae0
MIT License
1.52k stars 834 forks source link

Add export to ONNX #161

Closed SthPhoenix closed 2 years ago

SthPhoenix commented 2 years ago

Describe the changes you've made

Add script model2onnx.py to export keras model to ONNX

Describe if there is any unusual behaviour of your code(Write NA if there isn't)

NA

Additional context (OPTIONAL)

With proposed script it's possible to export model to ONNX which can be used in ONNXRuntime or TensorRT, both were tested.

Test plan (OPTIONAL)

Checklist