https://github.com/spacewalk01/tensorrt-yolov9

Cpp and python implementation of YOLOv9 using TensorRT API
https://github.com/spacewalk01/tensorrt-yolov9

cpp object-detection tensorrt yolo yolov5 yolov9

Last synced: 3 months ago
JSON representation

Cpp and python implementation of YOLOv9 using TensorRT API

• Host: GitHub
• URL: https://github.com/spacewalk01/tensorrt-yolov9
• Owner: spacewalk01
• License: mit
• Created: 2024-02-23T05:14:44.000Z (12 months ago)
• Default Branch: main
• Last Pushed: 2024-03-14T03:40:15.000Z (11 months ago)
• Last Synced: 2024-08-01T03:18:41.531Z (6 months ago)
• Topics: cpp, object-detection, tensorrt, yolo, yolov5, yolov9
• Language: C++
• Homepage: https://github.com/WongKinYiu/yolov9
• Size: 15.9 MB
• Stars: 108
• Watchers: 2
• Forks: 18
• Open Issues: 2
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        


TensorRT-YOLOv9

===========================

[![python](https://img.shields.io/badge/python-3.10.12-green)](https://www.python.org/downloads/release/python-31012/)

[![cuda](https://img.shields.io/badge/cuda-11.6-green)](https://developer.nvidia.com/cuda-downloads)

[![trt](https://img.shields.io/badge/TRT-8.6-green)](https://developer.nvidia.com/tensorrt)

[![mit](https://img.shields.io/badge/license-MIT-blue)](https://github.com/spacewalk01/TensorRT-YOLOv9/tree/main?tab=MIT-1-ov-file#readme)



 

This repo hosts a C++ and python implementation of the [YOLOv9](https://github.com/WongKinYiu/yolov9) state of the art object detection model, leveraging the TensorRT API for efficient real-time inference.



  

   



## 🚀 Usage

#### Python

``` shell

cd /python

python yolov9_trt.py --engine yolov9-c.engine --data images --outdir output

```

#### C++

``` shell

cd /build/release

# infer an image

yolov9-tensorrt.exe yolov9-c.engine test.jpg

# infer a folder(images)

yolov9-tensorrt.exe yolov9-c.engine data

# infer a video

yolov9-tensorrt.exe yolov9-c.engine test.mp4 # the video path

```

## 🛠️ Build

#### Python

The following command will install tensorrt for python:

``` shell

cd /python

pip install cuda-python

pip install tensorrt-8.6.0-cp310-none-win_amd64.whl

pip install opencv-python

```

#### C++

Refer to our [docs/INSTALL.md](https://github.com/spacewalk01/tensorrt-yolov9/blob/main/docs/INSTALL.md) for detailed installation instructions.

- Cuda preprocessing: [main branch](https://github.com/spacewalk01/TensorRT-YOLOv9/tree/main)

- Cpu preprocessing: [cpu-preprocessing branch](https://github.com/spacewalk01/TensorRT-YOLOv9/tree/cpu_preprocessing)

  

## 🖥️ Requirement

   - TensorRT

   - CUDA, CudaNN

   - C++ compiler with C++17 or higher support

   - CMake 3.14 or higher

   - OpenCV

## 🌱 Examples

Parameters for inference:

``` 

Confidence threshold : 0.2

NMS threshold : 0.3

Model : yolov9-e (with FP32)

```

Inference result:



   



Original image: https://www.flickr.com/photos/nicolelee/19041780


## 👏 Acknowledgement

This project is based on the following awesome projects:

- [YOLOv9](https://github.com/WongKinYiu/yolov9) - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information.

- [TensorRT](https://github.com/NVIDIA/TensorRT/tree/release/8.6/samples) - TensorRT samples and api documentation.

- [TensorRTx](https://github.com/wang-xinyu/tensorrtx) - Implementation of popular deep learning networks with TensorRT network definition API.

## 🔗 See also

- [Yolov9-Bytetrack](https://github.com/spacewalk01/yolov9-bytetrack-tensorrt) - Integration of YOLOv9 with ByteTracker using the TensorRT API.