Synopsis
dpuGetOutputTensorInHWCInt8
(
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
Get DPU Task’s output Tensor and store its INT8 type data 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