当前位置:首页 > 编程笔记 > 正文
已解决

DeepStream--测试TrafficCamNet检测模型

来自网友在路上 184884提问 提问时间:2023-11-21 17:38:34阅读次数: 84

最佳答案 问答题库848位专家为你答疑解惑

 模型地址:https://catalog.ngc.nvidia.com/orgs/nvidia/teams/tao/models/trafficcamnet/version

目前模型是nvidia的加密格式etlt。

nvinfer的配置

[property]
gpu-id=0
net-scale-factor=0.0039215697906911373
tlt-model-key=tlt_encode
tlt-encoded-model=trafficcamnet/resnet18_trafficcamnet_pruned.etlt
labelfile-path=labels_trafficnet.txt
int8-calib-file=trafficcamnet/trafficnet_int8.txt
model-engine-file=trafficcamnet/resnet18_trafficcamnet_pruned.etlt_b4_gpu0_int8.engine
input-dims=3;544;960;0
uff-input-blob-name=input_1
batch-size=4
process-mode=1
model-color-format=0
## 0=FP32, 1=INT8, 2=FP16 mode
network-mode=1
num-detected-classes=4
interval=0
gie-unique-id=1
output-blob-names=output_bbox/BiasAdd;output_cov/Sigmoid

[class-attrs-all]
pre-cluster-threshold=0.2
group-threshold=1
## Set eps=0.7 and minBoxes for cluster-mode=1(DBSCAN)
eps=0.2
#minBoxes=3

测试图片:/opt/nvidia/deepstream/deepstream/samples/streams/sample_720p.jpg

测试命令:

gst-launch-1.0 filesrc location=sample_720p.jpg ! jpegdec ! videoconvert ! video/x-raw,format=I420 ! nvvideoconvert ! video/x-raw\(memory:NVMM\),format=NV12 ! mux.sink_0 nvstreammux name=mux batch-size=1 width=1280 height=720 ! nvinfer config-file-path=./trafficamnet_config.txt ! nvvideoconvert ! video/x-raw\(memory:NVMM\),format=RGBA ! nvdsosd ! nvvideoconvert ! video/x-raw,format=I420 ! jpegenc ! filesink location=out.jpg

测试结果:

查看全文

99%的人还看了

猜你感兴趣

版权申明

本文"DeepStream--测试TrafficCamNet检测模型":http://eshow365.cn/6-41414-0.html 内容来自互联网,请自行判断内容的正确性。如有侵权请联系我们,立即删除!