dpuGetOutputTensorInCHWFP32() - 1.1 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2020-03-23
Version
1.1 English

Synopsis

dpuGetOutputTensorInCHWFP32
(
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 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