Fix YOLO-NAS / Small fixes

This commit is contained in:
Marcos Luciano
2023-06-16 14:16:51 -03:00
parent 936d1480c5
commit f8758d0e0e
10 changed files with 44 additions and 13 deletions

View File

@@ -0,0 +1,29 @@
[property]
gpu-id=0
net-scale-factor=1
model-color-format=0
onnx-file=yolo_nas_s_coco.onnx
model-engine-file=model_b1_gpu0_fp32.engine
#int8-calib-file=calib.table
labelfile-path=labels.txt
batch-size=1
network-mode=0
num-detected-classes=80
interval=0
gie-unique-id=1
process-mode=1
network-type=0
cluster-mode=2
maintain-aspect-ratio=1
symmetric-padding=0
#force-implicit-batch-dim=1
#workspace-size=1000
parse-bbox-func-name=NvDsInferParseYoloE
#parse-bbox-func-name=NvDsInferParseYoloECuda
custom-lib-path=nvdsinfer_custom_impl_Yolo/libnvdsinfer_custom_impl_Yolo.so
engine-create-func-name=NvDsInferYoloCudaEngineGet
[class-attrs-all]
nms-iou-threshold=0.45
pre-cluster-threshold=0.25
topk=300