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

13
customModels.md
View File

@@ -14,7 +14,7 @@ How to edit DeepStream files to your custom model
##
### 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)
* [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
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.
@@ -59,7 +59,14 @@ Note: config_infer_primary.txt uses cluster-mode=4 and NMS = 0.45 (via code) whe
### Compiling edited model
1. Check your CUDA version (nvcc --version)
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
```