Skip to content

This repo mainly contains documents on how to deploy a yolov5 model with tracking algorithm (Sort) on RK3588/RK3588s

Notifications You must be signed in to change notification settings

Rhyme0730/Yolov5-Sort-RKNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Run yolov5-sort on RK3588

Installation

Tips: If you want to integrate yolov5-sort with ROS/ROS2, using conda env is not recommended.

We use ROS2 Humble on Ubuntu 22.04 with default python version 3.10. You should be concerned about the conflict between different package versions.

Step 1: Install RKNN-Toolkit-Lite and Runtime package

cd Installation
pip install rknn_toolkit_lite2-1.5.2-cp310-cp310-linux_aarch64.whl
cp librknnrt.so /usr/lib

Step 2: Install essential package for Sort

pip install -r requirements.txt

If there are errors, follow the error instruction to install other packages

Step 3: Connect the stereo camera and run yolov5-sort on RK3588|RK3588s

cd ..
cd Code
python3 main.py

If you want to check the NPU load, open another shell and run

sudo bash rkcat.sh

Parameters Setting

Name Description
OBJ_THRESH Threshold for detecting
NMS_THRESH Non-maximum Suppression
TPEs The number of threads
modelPath .rknn model

Reference

rknn-multi-threaded
rknn-multi-threaded-3588

About

This repo mainly contains documents on how to deploy a yolov5 model with tracking algorithm (Sort) on RK3588/RK3588s

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published