Closed BorisPolonsky closed 2 years ago
You can see the classification API documentation here: https://github.com/triton-inference-server/server/blob/main/docs/protocol/extension_classification.md.
Looks like the config is not picking up the label file correctly. The expected output should be as described in the documentation : ["{value}:{idx}:label-c", "{value}:{idx}:label-d", ...]
Can you share your complete response? Also not sure whether it should work just rename label to label.txt
?
Closing due to lack of activity
Is your feature request related to a problem? Please describe. I tried eliminating label look up code on the client side via
label_filename
inconfig.pbtxt
. Yet I can't find this in official document. I tried referring to issue #3467, yet the content in corresponding output tensors (by settingoutput.parameters['classification'].int64_param = FLAGS.classes
as suggested) contains nothing defined in the label_file. For instance, given fileconfig.pbtxt
file
label
and the following script,
I got following output:
It tuns out that by specifying
output.parameters['classification'].int64_param
as suggest, the model server will sort numbers among the last dimension and encode output as["{value}:{idx}", ...]
instead of some["label-c", "label-d", ...]
sorted by their corresponding values, which is the expected behavior.Describe the solution you'd like
label_filename
configuration.Describe alternatives you've considered A clear and concise description of any alternative solutions or features you've considered.
Additional context Server version: nvcr.io/nvidia/tritonserver:21.09-py3