FaSTTUBe/GST-Tensordecoder-ov_ep
37
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tfliteinference: initialize means and stddevs arrays appropriately

Browse Source 
Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/9189>
This commit is contained in:
raghu447 2025-06-06 18:34:02 +05:30 committed by GStreamer Marge Bot
parent 0f7ee14a1d
commit a5586a50c2
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
subprojects/gst-plugins-bad/ext/tflite/gsttfliteinference.c
View File

@ -577,6 +577,10 @@ gst_tflite_inference_start (GstBaseTransform * trans)
if (num_means != priv->channels) { if (num_means != priv->channels) {
priv->means = g_renew (gdouble, priv->means, priv->channels); priv->means = g_renew (gdouble, priv->means, priv->channels);
/* initialize means array to zeroes */
if (num_means == 0) {
priv->means[0] = 0;
}
for (guint j = 1; j < priv->channels; j++) for (guint j = 1; j < priv->channels; j++)
priv->means[j] = priv->means[0]; priv->means[j] = priv->means[0];
} }
@ -586,6 +590,10 @@ gst_tflite_inference_start (GstBaseTransform * trans)
if (num_stddevs != priv->channels) { if (num_stddevs != priv->channels) {
priv->stddevs = g_renew (gdouble, priv->stddevs, priv->channels); priv->stddevs = g_renew (gdouble, priv->stddevs, priv->channels);
/* initialize stddevs array to ones */
if (num_stddevs == 0) {
priv->stddevs[0] = 1.0;
}
for (guint j = 1; j < priv->channels; j++) for (guint j = 1; j < priv->channels; j++)
priv->stddevs[j] = priv->stddevs[0]; priv->stddevs[j] = priv->stddevs[0];
} }