A real-time hand gesture recognition
Inference acceleration by Nvidia Tensorrt , the inference time dropped from 884ms to 9.3ms
- NVIDIA TensorRT SDK for inference
- MouseChannel Plugin for NVIDIA TensorRT SDK to use custom operator in tensorrt
- InterWild Published at CVPR 2023 by Facebook (recognition of hand gestures 2023 SOTA)
has tested in Ubuntu23.10,by Tensorrt 8.6.1
Pytorch Model ➡️ ONNX ➡️ Tensorrt
using three custom operators
- 🌟MouseChannelInverse🌟 : Matrix Inverse by Cuda
- 🌟MouseChannelSelect🌟 : Handling dynamic tensor size which is not allowed in native static compute graph
- 🌟MouseChannelSmallAngle🌟: Integrated some operations in pytorch and implemented them in cuda
In Pytorch Model Three custom operations are registered. In Tensorrt Use plugins to implement these custom operators
native pytorch | Tensorrt using float32 | |
---|---|---|
time of inference | 884.3ms | 9.31ms (9498%🚀) |
accuracy error(case1) | 0% | 1.1% |
accuracy error (case2) | 0% | 0.3% |
Real-time hand recognition,is able to generate Hand Joints Image and generate Hand Mesh
tested on
- real-time hand-joint-inference
- real-time animated hand-mesh
tested on
- real-time hand-joint-inference
- real-time animated hand-mesh