Support for YOLOv5 4.0

Added support for YOLOv5 4.0
2021-05-03 21:01:05 -03:00
parent 170f9cac74
commit ae849d2f8b
10 changed files with 995 additions and 2 deletions
--- a/external/yolov5-4.0/config_infer_primary.txt
+++ b/external/yolov5-4.0/config_infer_primary.txt
@@ -0,0 +1,18 @@
+[property]
+gpu-id=0
+net-scale-factor=0.0039215697906911373
+model-color-format=0
+model-engine-file=yolov5s.engine
+labelfile-path=labels.txt
+num-detected-classes=80
+interval=0
+gie-unique-id=1
+process-mode=1
+network-type=0
+cluster-mode=4
+maintain-aspect-ratio=0
+parse-bbox-func-name=NvDsInferParseCustomYoloV5
+custom-lib-path=nvdsinfer_custom_impl_Yolo/libnvdsinfer_custom_impl_Yolo.so
+
+[class-attrs-all]
+pre-cluster-threshold=0.25