Synopsis
dpuGetOutputTensorInCHWInt8
(
task,
nodeName,
int size,
idx = 0
)Arguments
- task
- The ctypes pointer to DPU Task.
- size
- The string DPU Node's name.
- 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 CHW (Channel*Height*Width).
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