Synopsis
dpuGetOutputTensorInHWCFP32
(
task,
nodeName,
int size,
idx = 0
)Arguments
- task
- The ctypes pointer to DPU Task.
- nodeName
- The string DPU Node's name.
- size
- The size (in Bytes) of output data to be stored.
- idx
- The index of a single output tensor for the Node, with default value of 0.
Description
Convert the data of DPU Task’s output Tensor from INT8 to float32, and store into CPU memory buffer under the layout of HWC (Height*Width*Channel).
Returns
NumPy array to hold the output data. Its size is zero in case of any error.
Include File
n2cube
Availability
Vitis AI v1.0