thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update to DeepStream 5.1

Browse Source
This commit is contained in:
Marcos Luciano
2021-02-28 19:14:03 -03:00
parent c69f50a75e
commit d0f0b66a1c
4 changed files with 44 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
YOLOv5.md
View File

@@ -1,5 +1,5 @@
# YOLOv5 # YOLOv5
NVIDIA DeepStream SDK 5.0.1 configuration for YOLOv5 models NVIDIA DeepStream SDK 5.1 configuration for YOLOv5 models
Thanks [DanaHan](https://github.com/DanaHan/Yolov5-in-Deepstream-5.0), [wang-xinyu](https://github.com/wang-xinyu/tensorrtx) and [Ultralytics](https://github.com/ultralytics/yolov5) Thanks [DanaHan](https://github.com/DanaHan/Yolov5-in-Deepstream-5.0), [wang-xinyu](https://github.com/wang-xinyu/tensorrtx) and [Ultralytics](https://github.com/ultralytics/yolov5)
@@ -153,8 +153,8 @@ sudo ./yolov5 -s
5. Create a custom yolo folder and copy generated files (example for YOLOv5s) 5. Create a custom yolo folder and copy generated files (example for YOLOv5s)
``` ```
mkdir /opt/nvidia/deepstream/deepstream-5.0/sources/yolo mkdir /opt/nvidia/deepstream/deepstream-5.1/sources/yolo
cp yolov5s.engine /opt/nvidia/deepstream/deepstream-5.0/sources/yolo/yolov5s.engine cp yolov5s.engine /opt/nvidia/deepstream/deepstream-5.1/sources/yolo/yolov5s.engine
``` ```
<br /> <br />
@@ -176,14 +176,22 @@ Edit yolov5.cpp file before compile if you want to change this parameters.
### Compile nvdsinfer_custom_impl_Yolo ### Compile nvdsinfer_custom_impl_Yolo
1. Run command 1. Run command
``` ```
sudo chmod -R 777 /opt/nvidia/deepstream/deepstream-5.0/sources/ sudo chmod -R 777 /opt/nvidia/deepstream/deepstream-5.1/sources/
``` ```
2. Donwload [my external/yolov5 folder](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/external/yolov5) and move files to created yolo folder 2. Donwload [my external/yolov5 folder](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/external/yolov5) and move files to created yolo folder
3. Compile lib 3. Compile lib
* x86 platform
``` ```
cd /opt/nvidia/deepstream/deepstream-5.0/sources/yolo cd /opt/nvidia/deepstream/deepstream-5.1/sources/yolo
CUDA_VER=11.1 make -C nvdsinfer_custom_impl_Yolo
```
* Jetson platform
```
cd /opt/nvidia/deepstream/deepstream-5.1/sources/yolo
CUDA_VER=10.2 make -C nvdsinfer_custom_impl_Yolo CUDA_VER=10.2 make -C nvdsinfer_custom_impl_Yolo
``` ```

13
customModels.md
View File

@@ -14,7 +14,7 @@ How to edit DeepStream files to your custom model
## ##
### Requirements ### Requirements
* [NVIDIA DeepStream SDK 5.0.1](https://developer.nvidia.com/deepstream-sdk) * [NVIDIA DeepStream SDK 5.1](https://developer.nvidia.com/deepstream-sdk)
* [DeepStream-Yolo Native](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/native) * [DeepStream-Yolo Native](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/native)
* [Pre-treined YOLO model](https://github.com/AlexeyAB/darknet) * [Pre-treined YOLO model](https://github.com/AlexeyAB/darknet)
@@ -23,7 +23,7 @@ How to edit DeepStream files to your custom model
### Editing default model ### Editing default model
1. Run command 1. Run command
``` ```
sudo chmod -R 777 /opt/nvidia/deepstream/deepstream-5.0/sources/ sudo chmod -R 777 /opt/nvidia/deepstream/deepstream-5.1/sources/
``` ```
2. Download [my native folder](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/native), rename to yolo and move to your deepstream/sources folder. 2. Download [my native folder](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/native), rename to yolo and move to your deepstream/sources folder.
@@ -59,7 +59,14 @@ Note: config_infer_primary.txt uses cluster-mode=4 and NMS = 0.45 (via code) whe
### Compiling edited model ### Compiling edited model
1. Check your CUDA version (nvcc --version) 1. Check your CUDA version (nvcc --version)
2. Go to deepstream/sources/yolo directory 2. Go to deepstream/sources/yolo directory
3. Type command (example for CUDA 10.2 version): 3. Type command to compile:
* x86 platform
```
CUDA_VER=11.1 make -C nvdsinfer_custom_impl_Yolo
```
* Jetson platform
``` ```
CUDA_VER=10.2 make -C nvdsinfer_custom_impl_Yolo CUDA_VER=10.2 make -C nvdsinfer_custom_impl_Yolo
``` ```

9
multipleInferences.md
View File

@@ -44,7 +44,14 @@ Note: 2 = sgie1, 3 = sgie2, 4 = sgie3, etc
### Compiling edited models ### Compiling edited models
1. Check your CUDA version (nvcc --version) 1. Check your CUDA version (nvcc --version)
2. Go to inference directory. 2. Go to inference directory.
3. Type command (example for CUDA 10.2 version): 3. Type command to compile:
* x86 platform
```
CUDA_VER=11.1 make -C nvdsinfer_custom_impl_Yolo
```
* Jetson platform
``` ```
CUDA_VER=10.2 make -C nvdsinfer_custom_impl_Yolo CUDA_VER=10.2 make -C nvdsinfer_custom_impl_Yolo
``` ```

18
readme.md
View File

@@ -1,5 +1,5 @@
# DeepStream-Yolo # DeepStream-Yolo
NVIDIA DeepStream SDK 5.0.1 configuration for YOLO models NVIDIA DeepStream SDK 5.1 configuration for YOLO models
## ##
@@ -47,13 +47,13 @@ Request
## ##
### Requirements ### Requirements
* [NVIDIA DeepStream SDK 5.0.1](https://developer.nvidia.com/deepstream-sdk) * [NVIDIA DeepStream SDK 5.1](https://developer.nvidia.com/deepstream-sdk)
* [DeepStream-Yolo Native](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/native) (for Darknet YOLO based models) * [DeepStream-Yolo Native](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/native) (for Darknet YOLO based models)
* [DeepStream-Yolo External](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/external) (for PyTorch YOLOv5 based model) * [DeepStream-Yolo External](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/external) (for PyTorch YOLOv5 based model)
## ##
### mAP/FPS comparison between models ### mAP/FPS comparison between models (OUTDATED)
DeepStream SDK YOLOv4: https://youtu.be/Qi_F_IYpuFQ DeepStream SDK YOLOv4: https://youtu.be/Qi_F_IYpuFQ
@@ -215,7 +215,7 @@ pre-cluster-threshold = 0.25 (CONF_THRESH)
Run command Run command
``` ```
sudo chmod -R 777 /opt/nvidia/deepstream/deepstream-5.0/sources/ sudo chmod -R 777 /opt/nvidia/deepstream/deepstream-5.1/sources/
``` ```
Download [my native folder](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/native), rename to yolo and move to your deepstream/sources folder. Download [my native folder](https://github.com/marcoslucianops/DeepStream-Yolo/tree/master/native), rename to yolo and move to your deepstream/sources folder.
@@ -239,8 +239,16 @@ Download cfg and weights files from your model and move to deepstream/sources/yo
Compile Compile
* x86 platform
``` ```
cd /opt/nvidia/deepstream/deepstream-5.0/sources/yolo cd /opt/nvidia/deepstream/deepstream-5.1/sources/yolo
CUDA_VER=11.1 make -C nvdsinfer_custom_impl_Yolo
```
* Jetson platform
```
cd /opt/nvidia/deepstream/deepstream-5.1/sources/yolo
CUDA_VER=10.2 make -C nvdsinfer_custom_impl_Yolo CUDA_VER=10.2 make -C nvdsinfer_custom_impl_Yolo
``` ```