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
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)
@@ -153,8 +153,8 @@ sudo ./yolov5 -s
5. Create a custom yolo folder and copy generated files (example for YOLOv5s)
```
mkdir /opt/nvidia/deepstream/deepstream-5.0/sources/yolo
cp yolov5s.engine /opt/nvidia/deepstream/deepstream-5.0/sources/yolo/yolov5s.engine
mkdir /opt/nvidia/deepstream/deepstream-5.1/sources/yolo
cp yolov5s.engine /opt/nvidia/deepstream/deepstream-5.1/sources/yolo/yolov5s.engine
```
<br />
@@ -176,14 +176,22 @@ Edit yolov5.cpp file before compile if you want to change this parameters.
### Compile nvdsinfer_custom_impl_Yolo
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
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
```