In nvinfer plugin, we have the options to set the inference model output confidence threshold (pre-cluster-threshold
) and the threshold after clustering (post-cluster-threshold
).
They can be set for all the classes (class-attrs-all
) or refined for a specific model class (class-attrs-<class_ID>
starting from ID 0):
[class-attrs-all]
pre-cluster-threshold=0.5
post-cluster-threshold=0.5
[class-attrs-0]
pre-cluster-threshold=0.3
post-cluster-threshold=0.5
[class-attrs-1]
pre-cluster-threshold=0.2
post-cluster-threshold=0.5
On nvinferserver plugin the pre_threshold
can be found in per_class_params
, which is an optional map<int32, PerClassParams>
:
per_class_params {
{ key: 1, value { pre_threshold : 0.4} },
{ key: 2, value { pre_threshold : 0.5} }
}
Doubt #1: Is there some way to set the pre_threshold
just once for all the model’s classes?
Doubt #2: Does the class ID integer starts at 0
just like nvinfer
, or starts at 1
as we can find in the nvinferserver
example?
If some cluster type is enabled, I can find on each a threshold field that according to the descriptions seems to perform the same action as the previously referred pre_threshold
:
NMS
confidence_threshold
- Detection score lesser than this threshold would be rejected
DbScan
pre_threshold
- Detection score lesser than this threshold would be rejected before DBSCAN clustering
GroupRectangle
confidence_threshold
- Detection score lesser than this threshold would be rejected
SimpleCluster
threshold
- Detection score lesser than this threshold would be rejected
Doubt #3: in practice the functionality of those fields produce the same effect as pre_threshold
from DetectionParams
?
Doubt #4: where can we find the equivalent for nvinfer
’s post-cluster-threshold
in nvinferserver
?