mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-13 04:41:16 +00:00
onnx: fix formating
Code alignement was not alway consistent Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/7205>
This commit is contained in:
parent
43edff4e29
commit
4ff98d66bb
1 changed files with 16 additions and 13 deletions
|
@ -324,7 +324,8 @@ GstOnnxClient::GstOnnxClient (GstElement *debug_parent):debug_parent(debug_paren
|
|||
}
|
||||
|
||||
// copy tensor data to a GstTensorMeta
|
||||
GstTensorMeta *GstOnnxClient::copy_tensors_to_meta (std::vector < Ort::Value >
|
||||
GstTensorMeta *
|
||||
GstOnnxClient::copy_tensors_to_meta (std::vector < Ort::Value >
|
||||
&outputs, GstBuffer * buffer)
|
||||
{
|
||||
size_t num_tensors = outputNamesRaw.size ();
|
||||
|
@ -372,7 +373,8 @@ GstOnnxClient::GstOnnxClient (GstElement *debug_parent):debug_parent(debug_paren
|
|||
buffer_size);
|
||||
tensor->data_type = GST_TENSOR_TYPE_INT32;
|
||||
} else {
|
||||
GST_ERROR_OBJECT (debug_parent, "Output tensor is not FLOAT32 or INT32, not supported");
|
||||
GST_ERROR_OBJECT (debug_parent,
|
||||
"Output tensor is not FLOAT32 or INT32, not supported");
|
||||
gst_buffer_remove_meta (buffer, (GstMeta *) tmeta);
|
||||
return NULL;
|
||||
}
|
||||
|
@ -383,7 +385,8 @@ GstOnnxClient::GstOnnxClient (GstElement *debug_parent):debug_parent(debug_paren
|
|||
}
|
||||
|
||||
std::vector < Ort::Value > GstOnnxClient::run (uint8_t * img_data,
|
||||
GstVideoInfo vinfo) {
|
||||
GstVideoInfo vinfo)
|
||||
{
|
||||
std::vector < Ort::Value > modelOutput;
|
||||
doRun (img_data, vinfo, modelOutput);
|
||||
|
||||
|
|
Loading…
Reference in a new issue