dpuGetOutputTensorInHWCFP32() - 1.4 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2021-07-22
Version
1.4 English

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