-
Notifications
You must be signed in to change notification settings - Fork 219
Add C++ ONNX Demo #76
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: develop
Are you sure you want to change the base?
Conversation
Hey @sctrueew, really cool work! Would love to merge. Can you sign the CLA? |
Yes, I have agreed the CLA |
Hi @sctrueew, thanks for accepting the CLA! Would it be possible for you to add a README in |
Hi @SkalskiP, I have added a README file to the project. Please let me know if you need any further changes or additional information. |
…DETR_ONNX.cpp for better structure and maintainability
This PR introduces a C++ demo for the RF-DETR model, allowing users to perform real-time object detection using an ONNX model. The demo supports various input sources, including images, videos, and live camera streams, with optional CUDA acceleration.
Key Features:
✅ Loads an RF-DETR model in ONNX format
✅ Supports image, video, and live camera inference
✅ Enables CPU and CUDA (GPU) execution
✅ Configurable confidence threshold for detections
✅ Outputs annotated images/videos with detected objects
✅ Uses COCO class labels for object recognition
Run Examples:
🔹 Image Inference:
Detect objects in a static image and save the output:
🔹 Video Inference:
Process a video file and save the annotated output:
🔹 Live Camera Inference (Default ID 0):
Run inference on the default webcam (ID 0) with GPU acceleration:
🔹 Live Camera Inference (Specific Camera ID 1):
Run inference on a specific camera (ID 1):
🔹 Get Help & Available Options:
./main --help
Dependencies: