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

Update YOLOv5-5.0.md

Browse Source
This commit is contained in:
Marcos Luciano
2021-05-09 18:56:09 -03:00
committed by GitHub
parent 6c01b0faf2
commit 702a57d8b0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
YOLOv5-5.0.md
View File

@@ -36,7 +36,7 @@ pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision
cd torchvision cd torchvision
export BUILD_VERSION=0.9.0 export BUILD_VERSION=0.9.0
python3 setup.py install --user python3 setup.py install --user
``` ```
@@ -99,7 +99,7 @@ cp yolov5s.engine /opt/nvidia/deepstream/deepstream-5.1/sources/yolo/yolov5s.eng
Note: by default, yolov5 script generate model with batch size = 1 and FP16 mode. Note: by default, yolov5 script generate model with batch size = 1 and FP16 mode.
``` ```
#define USE_FP32 // set USE_INT8 or USE_FP16 or USE_FP32 #define USE_FP16 // set USE_INT8 or USE_FP16 or USE_FP32
#define DEVICE 0 // GPU id #define DEVICE 0 // GPU id
#define NMS_THRESH 0.4 #define NMS_THRESH 0.4
#define CONF_THRESH 0.5 #define CONF_THRESH 0.5