ScorpioPeng/3D_Object_detection_with_ONNX

3D_object_detection_with_depth_camera

unfinished project

Installation

pip install -r requirements.txt

ONNX model

Download the models from here and here, and place them in the models folder. For converting the original Pytorch model to ONNX, check the following branch: https://github.com/ibaiGorordo/CenterSnap/tree/convert_onnx

References

• ONNX 6DPose and ShapeEstimation = https://github.com/ibaiGorordo/ONNX-CenterSnap-6D-Pose-and-Shape-Estimation
• CenterSnap: https://github.com/zubair-irshad/CenterSnap
• Original paper: https://arxiv.org/abs/2203.01929
• Modified CenterSnap for conversion: https://github.com/ibaiGorordo/CenterSnap/tree/convert_onnx
• Depthai library: https://github.com/luxonis/depthai-python
• PyRealsense2 library: https://github.com/IntelRealSense/librealsense
• pyKinectAzure: https://github.com/ibaiGorordo/pyKinectAzure