Skip to content

YOLOv4 accelerated wtih TensorRT and multi-stream input using Deepstream

License

Notifications You must be signed in to change notification settings

kingardor/YOLOv4-Deepstream

Repository files navigation

Deepstream 5.1 YOLOv4 App

This Deepstream application showcases YOLOv4 running at high FPS throughput!

FPS

P.S - Click the gif to watch the entire video!

Index

  1. Deepstream Setup
    1. Install System Dependencies
    2. Install Deepstream
  2. Running the Application
    1. Clone the repository
    2. Download the weights file
    3. Build the application
    4. Run with different input sources
  3. Citations

Deepstream Setup

This post assumes you have a fully functional Jetson device. If not, you can refer the documentation here.

1. Install System Dependencies

sudo apt install \
libssl1.0.0 \
libgstreamer1.0-0 \
gstreamer1.0-tools \
gstreamer1.0-plugins-good \
gstreamer1.0-plugins-bad \
gstreamer1.0-plugins-ugly \
gstreamer1.0-libav \
libgstrtspserver-1.0-0 \
libjansson4=2.11-1

2. Install Deepstream

Download the DeepStream 5.1 Jetson Debian package deepstream-5.1_5.1.0-1_arm64.deb, to the Jetson device from here. Then enter the command:

sudo apt install deepstream-5.1_5.1.0-1_arm64.deb

For more information, go to the get started page of Deepstream here.

Running the Application

1. Clone the repository

This is a straightforward step, however, if you are new to git, I recommend glancing threw the steps.

First, install git

sudo apt install git

Next, clone the repository

# Using HTTPS
https://github.com/aj-ames/YOLOv4-Deepstream.git
# Using SSH
git@github.com:aj-ames/YOLOv4-Deepstream.git

2. Download the weights file

Download the weights file from google-drive and place it in models/YOLOv4 directory.

3. Build the application

First, build the application by running the following command:

make clean && make -j$(nproc)

This will generate the binary called ds-yolo. This is a one-time step and you need to do this only when you make source-code changes.

4. Run with different input sources

Next, create a file called inputsources.txt and paste the path of videos or rtsp url.

file:///home/astr1x/Videos/sample.mp4
rtsp://admin:admin%40123@192.168.1.1:554/stream

Now, run the application by running the following command:

./ds-yolo

Citations

About

YOLOv4 accelerated wtih TensorRT and multi-stream input using Deepstream

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published