TLT-deepstream sample SSD app problems : I used SSD sample with official etlt model can run successfully, but my etlt can’t

command: ./deepstream-custom -c pgie_ssd_tlt_config.txt -i $DS_SRC_PATH/samples/streams/sample_720p.h264 -d
Warning: ‘input-dims’ parameter has been deprecated. Use ‘infer-dims’ instead.
Now playing: pgie_ssd_tlt_config.txt

Using winsys: x11
Opening in BLOCKING MODE
0:00:00.204557041 13589 0x559678a670 INFO nvinfer gstnvinfer.cpp:619:gst_nvinfer_logger: NvDsInferContext[UID 1]: Info from NvDsInferContextImpl::buildModel() <nvdsinfer_context_impl.cpp:1715> [UID = 1]: Trying to create engine from model files
INFO: [TRT]: Detected 1 inputs and 2 output network tensors.
#assertion/home/nvidia/Public/TensorRT/plugin/nmsPlugin/nmsPlugin.cpp,246
Aborted (core dumped)

pgie_ssd_tlt_config.txt:

[property]
gpu-id=0
net-scale-factor=1.0
offsets=103.939;116.779;123.68
model-color-format=1
#kitti classes
labelfile-path=./nvdsinfer_customparser_ssd_tlt/ssd_labels.txt
#tlt-encoded-model=./models/ssd/ssd_resnet18.etlt
#tlt-model-key=nvidia_tlt
tlt-encoded-model=./models/ssd/ssd_resnet18_fp32_epoch_050.etlt
tlt-model-key=ZDVhYWl****************************************************************
uff-input-dims=3;544;960;0
uff-input-blob-name=Input
batch-size=1
#0=FP32, 1=INT8, 2=FP16 mode
network-mode=0
#num of my classes
num-detected-classes=3
interval=0
gie-unique-id=1
is-classifier=0
#network-type=0
output-blob-names=NMS
parse-bbox-func-name=NvDsInferParseCustomSSDTLT
custom-lib-path=./nvdsinfer_customparser_ssd_tlt/libnvds_infercustomparser_ssd_tlt.so

[class-attrs-all]
pre-cluster-threshold=0.3
roi-top-offset=0
roi-bottom-offset=0
detected-min-w=0
detected-min-h=0
detected-max-w=0
detected-max-h=0

I used TLT SSD to train Kitti dataset, and there are three classes:
car
cyclist
pedestrian

Refer to Fail to create SSD engine

1 Like

Thank you. I have solved this problem. I have modified this parameter before, but somehow it is wrong. It may be that there was a problem with other configuration at that time.
image