I have a hef model that takes an input tensor with dimensions (10, 8, 4096) (H, W, C) In my gstreamer pipeline the upstream element sets the caps as (see gstreamer pipeline image below)
Good question! The GStreamer element is actually built pretty tightly around standard video inputs (mostly RGB), so that’s more of a design constraint than anything to do with the Python API itself.
For non-standard tensor shapes like what you’re working with, the Python API is definitely the way to go – and it sounds like you’ve already got that working, which is great!
I’ll check with our R&D team to see if we can add more flexibility to the GStreamer element in an upcoming release, but for now I’d recommend sticking with the Python API (or the C++ API if you need that level of control).