From e84dd6b975cfeeab6d9e09a7e49a6af33272c5d9 Mon Sep 17 00:00:00 2001 From: Varun Praveen Date: Tue, 5 Oct 2021 11:21:07 -0500 Subject: [PATCH] [Configs] Update the DNV2 config files --- .../config_infer_primary_dashcamnet.txt | 42 ++++++++++--------- .../config_infer_primary_facedetectir.txt | 35 ++++++++-------- .../config_infer_primary_peoplenet.txt | 25 +++++------ .../config_infer_primary_trafficcamnet.txt | 40 ++++++++++-------- 4 files changed, 75 insertions(+), 67 deletions(-) diff --git a/deepstream_app_tao_configs/config_infer_primary_dashcamnet.txt b/deepstream_app_tao_configs/config_infer_primary_dashcamnet.txt index ea475bc..be8c7d5 100644 --- a/deepstream_app_tao_configs/config_infer_primary_dashcamnet.txt +++ b/deepstream_app_tao_configs/config_infer_primary_dashcamnet.txt @@ -39,30 +39,32 @@ num-detected-classes=4 interval=0 gie-unique-id=1 output-blob-names=output_bbox/BiasAdd;output_cov/Sigmoid -cluster-mode=2 +# Clustering mode set to DBSCAN +cluster-mode=1 #Use the config params below for dbscan clustering mode -#[class-attrs-all] -#detected-min-w=4 -#detected-min-h=4 -#minBoxes=3 -#eps=0.7 +[class-attrs-all] +detected-min-w=4 +detected-min-h=4 +minBoxes=3 +eps=0.7 +pre-cluster-threshold=0.05 #Use the config params below for NMS clustering mode -[class-attrs-all] -topk=20 -nms-iou-threshold=0.5 -pre-cluster-threshold=0.2 -roi-top-offset=0 -roi-bottom-offset=0 +# [class-attrs-all] +# topk=20 +# nms-iou-threshold=0.5 +# pre-cluster-threshold=0.2 +# roi-top-offset=0 +# roi-bottom-offset=0 ## Per class configurations -[class-attrs-0] -topk=20 -nms-iou-threshold=0.5 -pre-cluster-threshold=0.4 +# [class-attrs-0] +# topk=20 +# nms-iou-threshold=0.5 +# pre-cluster-threshold=0.4 -#[class-attrs-1] -#pre-cluster-threshold=0.05 -#eps=0.7 -#dbscan-min-score=0.5 +# [class-attrs-1] +# pre-cluster-threshold=0.05 +# eps=0.7 +# dbscan-min-score=0.5 diff --git a/deepstream_app_tao_configs/config_infer_primary_facedetectir.txt b/deepstream_app_tao_configs/config_infer_primary_facedetectir.txt index 0cc5f37..170149b 100644 --- a/deepstream_app_tao_configs/config_infer_primary_facedetectir.txt +++ b/deepstream_app_tao_configs/config_infer_primary_facedetectir.txt @@ -39,28 +39,29 @@ num-detected-classes=1 interval=0 gie-unique-id=1 output-blob-names=output_bbox/BiasAdd;output_cov/Sigmoid -cluster-mode=2 +cluster-mode=1 #Use the config params below for dbscan clustering mode -#[class-attrs-all] -#detected-min-w=4 -#detected-min-h=4 -#minBoxes=3 -#eps=0.7 +[class-attrs-all] +pre-cluster-threshold=0.05 +detected-min-w=4 +detected-min-h=4 +minBoxes=3 +eps=0.7 #Use the config params below for NMS clustering mode -[class-attrs-all] -topk=20 -nms-iou-threshold=0.5 -pre-cluster-threshold=0.2 +# [class-attrs-all] +# topk=20 +# nms-iou-threshold=0.5 +# pre-cluster-threshold=0.2 ## Per class configurations -[class-attrs-0] -topk=20 -nms-iou-threshold=0.5 -pre-cluster-threshold=0.4 +# [class-attrs-0] +# topk=20 +# nms-iou-threshold=0.5 +# pre-cluster-threshold=0.4 #[class-attrs-1] -#pre-cluster-threshold=0.05 -#eps=0.7 -#dbscan-min-score=0.5 +# pre-cluster-threshold=0.05 +# eps=0.7 +# dbscan-min-score=0.5 diff --git a/deepstream_app_tao_configs/config_infer_primary_peoplenet.txt b/deepstream_app_tao_configs/config_infer_primary_peoplenet.txt index eeb1aea..1083730 100644 --- a/deepstream_app_tao_configs/config_infer_primary_peoplenet.txt +++ b/deepstream_app_tao_configs/config_infer_primary_peoplenet.txt @@ -34,7 +34,7 @@ batch-size=1 process-mode=1 model-color-format=0 ## 0=FP32, 1=INT8, 2=FP16 mode -network-mode=1 +network-mode=2 num-detected-classes=3 cluster-mode=2 interval=0 @@ -43,22 +43,23 @@ output-blob-names=output_bbox/BiasAdd;output_cov/Sigmoid #Use the config params below for dbscan clustering mode #[class-attrs-all] -#detected-min-w=4 -#detected-min-h=4 -#minBoxes=3 -#eps=0.7 +detected-min-w=4 +detected-min-h=4 +minBoxes=3 +eps=0.7 +pre-cluster-threshold=0.4 #Use the config params below for NMS clustering mode [class-attrs-all] -topk=20 -nms-iou-threshold=0.5 -pre-cluster-threshold=0.2 +# topk=20 +# nms-iou-threshold=0.5 +# pre-cluster-threshold=0.2 ## Per class configurations -[class-attrs-0] -topk=20 -nms-iou-threshold=0.5 -pre-cluster-threshold=0.4 +# [class-attrs-0] +# topk=20 +# nms-iou-threshold=0.5 +# pre-cluster-threshold=0.4 #[class-attrs-1] #pre-cluster-threshold=0.05 diff --git a/deepstream_app_tao_configs/config_infer_primary_trafficcamnet.txt b/deepstream_app_tao_configs/config_infer_primary_trafficcamnet.txt index 5ef0370..74d9ab2 100644 --- a/deepstream_app_tao_configs/config_infer_primary_trafficcamnet.txt +++ b/deepstream_app_tao_configs/config_infer_primary_trafficcamnet.txt @@ -39,28 +39,32 @@ num-detected-classes=4 interval=0 gie-unique-id=1 output-blob-names=output_bbox/BiasAdd;output_cov/Sigmoid -cluster-mode=2 +# Clustering mode set to DBSCAN +cluster-mode=1 #Use the config params below for dbscan clustering mode -#[class-attrs-all] -#detected-min-w=4 -#detected-min-h=4 -#minBoxes=3 -#eps=0.7 +[class-attrs-all] +detected-min-w=4 +detected-min-h=4 +minBoxes=3 +eps=0.7 +pre-cluster-threshold=0.05 #Use the config params below for NMS clustering mode -[class-attrs-all] -topk=20 -nms-iou-threshold=0.5 -pre-cluster-threshold=0.2 +# [class-attrs-all] +# topk=20 +# nms-iou-threshold=0.5 +# pre-cluster-threshold=0.2 +# roi-top-offset=0 +# roi-bottom-offset=0 ## Per class configurations -[class-attrs-0] -topk=20 -nms-iou-threshold=0.5 -pre-cluster-threshold=0.4 +# [class-attrs-0] +# topk=20 +# nms-iou-threshold=0.5 +# pre-cluster-threshold=0.4 -#[class-attrs-1] -#pre-cluster-threshold=0.05 -#eps=0.7 -#dbscan-min-score=0.5 +# [class-attrs-1] +# pre-cluster-threshold=0.05 +# eps=0.7 +# dbscan-min-score=0.5 \ No newline at end of file