Please provide complete information as applicable to your setup.
• Hardware Platform (Jetson / GPU)
My graphic card is RTX 3080.
• DeepStream Version
The Version is 6.0
• TensorRT Version
TensorRT Version on tags 21.08 is v8.0.1.6.
• NVIDIA GPU Driver Version (valid for GPU only)
The Driver version is 470.129.06.
• Issue Type( questions, new requirements, bugs)
Firstly I use nvidia/digits docker to train a caffe model and download model.
The model contains a deploy.prototxt and a snapshot_iter_3362.caffemodel .
The output layer name in deploy.prototxt is softmax.
So I executed the trtexec command which is
“./trtexec --deploy=model/deploy.prototxt --model=model/snapshot_iter_3362.caffemodel --output=softmax --batch=16 --saveEngine=deploy.engine”
Finally, in deepstream app secondary-gie0 settings, I modify the config-file as follows.
[property]
gpu-id=0
net-scale-factor=1.5
model-file=snapshot_iter_3362.caffemodel
proto-file=deploy.prototxt
model-engine-file=deploy.engine
mean-file=csc.ppm
labelfile-path=caffe_label.txt
force-implicit-batch-dim=1
batch-size=16
model-color-format=1
network-mode=0
is-classifier=1
process-mode=1
output-blob-names=softmax
classifier-async-mode=1
classifier-threshold=0.001
input-object-min-width=128
input-object-min-height=128
operate-on-gie-id=1
operate-on-class-ids=5;6
classifier-type=vehicletype
The deepstream app can execute well, but the result of sgie, class id is always 0.
I use opencv dnn to load caffe mode and prototxt, and get good inference resulit of class id.
In fact, no matter I use trtexec to produce engine or use deepstream app to produce snapshot_iter_3362.caffemodel_b16_gpu0_fp32.engine, the class id of inference result is always zero.
Thank you for your help and suggestion.
• How to reproduce the issue ? (This is for bugs. Including which sample app is using, the configuration files content, the command line used and other details for reproducing)
• Requirement details( This is for new requirement. Including the module name-for which plugin or for which sample application, the function description)