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

Update YOLOv5-4.0.md

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

2
YOLOv5-4.0.md
View File

@@ -109,7 +109,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