[Example] RTDETR-ONNXRuntime-Python (#18369)

This commit is contained in:
Semih Demirel 2024-12-24 14:18:34 +03:00 committed by GitHub
parent 12db1f3143
commit 5b76bed7d0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 257 additions and 0 deletions

View file

@ -0,0 +1,43 @@
# RTDETR - ONNX Runtime
This project implements RTDETR using ONNX Runtime.
## Installation
To run this project, you need to install the required dependencies. The following instructions will guide you through the installation process.
### Installing Required Dependencies
You can install the required dependencies by running the following command:
```bash
pip install -r requirements.txt
```
### Installing `onnxruntime-gpu`
If you have an NVIDIA GPU and want to leverage GPU acceleration, you can install the onnxruntime-gpu package using the following command:
```bash
pip install onnxruntime-gpu
```
Note: Make sure you have the appropriate GPU drivers installed on your system.
### Installing `onnxruntime` (CPU version)
If you don't have an NVIDIA GPU or prefer to use the CPU version of onnxruntime, you can install the onnxruntime package using the following command:
```bash
pip install onnxruntime
```
### Usage
After successfully installing the required packages, you can run the RTDETR implementation using the following command:
```bash
python main.py --model rtdetr-l.onnx --img image.jpg --conf-thres 0.5 --iou-thres 0.5
```
Make sure to replace rtdetr-l.onnx with the path to your RTDETR ONNX model file, image.jpg with the path to your input image, and adjust the confidence threshold (conf-thres) and IoU threshold (iou-thres) values as needed.