I have a SGIE classifier of which I’d like to obtain the embedding from the last layer using the Python API of Deepstream 5.0.1. I run TensorRT 7.2.1, by requirement for this model.
Inside my application (where I already get the correct bbox from my TensorRT detector, the raw frames, etc), I execute
obj_usr_meta_list = obj_meta.obj_user_meta_list if obj_usr_meta_list: user_meta_list=pyds.NvDsUserMeta.cast(obj_usr_meta_list.data) user_meta_data = pyds.NvDsInferTensorMeta.cast(user_meta_list.user_meta_data) output_layers_info = user_meta_data.output_layers_info(0) print(output_layers_info.layerName) print(output_layers_info.buffer) print(output_layers_info)
>>> <pyds.NvDsInferLayerInfo object at 0x7f83c818cf10>
It appears to me that my model executes correctly and I can’t think of any way of checking that it writes to this field.
I remember with the beta version that some structures that weren’t implemented would return
None, is this what’s happening here, or is it my model that is wrong?
I also tried the
out_buf_ptrs_host property of
pyds.NvDsInferTensorMeta with the same result.
My nvinfer config