ModelDataset
TensorDesc
功能
张量描述信息定义。
结构定义
struct TensorDesc {
size_t tensorSize;
std::string tensorName;
std::vector<int64_t> tensorDims;
};
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| tensorSize | 输出 | 张量的大小,单位为Byte。 |
| tensorName | 输出 | 张量名称。 |
| tensorDims | 输出 | 张量维度。 |
父主题: 通用数据结构
ModelDesc
功能
模型描述信息定义。
结构定义
struct ModelDesc {
std::vector<TensorDesc> inputTensors;
std::vector<TensorDesc> outputTensors;
std::vector<size_t> batchSizes;
bool dynamicBatch;
};
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| inputTensors | 输出 | 输入张量描述。 |
| outputTensors | 输出 | 输出张量描述。 |
| batchSizes | 输出 | batch大小。 |
| dynamicBatch | 输出 | 是否为动态Batch。 |
父主题: 通用数据结构
ModelDataset
功能
模型数据集定义。
结构定义
struct ModelDataset {
void* mdlDataPtr;
size_t dynamicBatchSize;
};
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| mdlDataPtr | 输入 | 模型数据集数据的内存地址。 |
| dynamicBatchSize | 输入 | 模型推理时的批量大小。 |
父主题: 通用数据结构
DvppDataInfo
功能
DVPP实体数据定义。
结构定义
struct DvppDataInfo {
uint32_t width = 0;
uint32_t height = 0;
uint32_t widthStride = 0;
uint32_t heightStride = 0;
MxbasePixelFormat format = MXBASE_PIXEL_FORMAT_YUV_SEMIPLANAR_420;
uint32_t frameId = 0;
uint32_t channelId = 0;
uint32_t dataSize = 0;
uint32_t outDataSize = 0;
uint32_t dataType = 0;
uint8_t* data = nullptr;
uint8_t* outData = nullptr;
uint32_t resizeWidth = 0;
uint32_t resizeHeight = 0;
std::string device = "host:0";
void (*destory)(void *) = nullptr;
};
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| width | 输入 | 原始图像宽。 |
| height | 输入 | 原始图像高。 |
| widthStride | 输入/输出 | 原始像片对齐后的宽。 |
| heightStride | 输入/输出 | 原始像片对齐后的高。 |
| format | 输入/输出 | 图像格式,默认值为“MXBASE_PIXEL_FORMAT_YUV_SEMIPLANAR_420”。 DVPP图像格式定义具体请参见MxbasePixelFormat。 |
| frameId | 输入 | 图像帧编号。 |
| channelId | 输入 | 图像通道数编号。 |
| dataSize | 输入/输出 | 图像数据大小,单位为Byte。 dataSize大小需要和图像实际大小一致,否则可能会导致程序出现coredump情况。 |
| outDataSize | 输入/输出 | 输出图像数据大小,单位为Byte。 outDataSize大小需要和输出图像大小一致,否则可能会导致程序出现coredump情况。 |
| dataType | 输入 | 图像的数据类型。 |
| data | 输入/输出 | 图像数据。 |
| outData | 输入/输出 | 预申请内存地址,主要用于存放视频解码后的图像数据。 |
| device | 输入 | 设备号。 |
| destory | 输入/输出 | 回调函数,用于释放该DVPP数据。 |
父主题: 通用数据结构
DvppImageInfo
功能
DVPP图像信息描述。
结构定义
struct DvppImageInfo {
enum PictureType {
PIXEL_FORMAT_ANY = 0,
PIXEL_FORMAT_JPEG = 1,
PIXEL_FORMAT_PNG = 2
};
const void* data;
uint32_t size;
PictureType pictureType;
};
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| data | 输入 | 图片数据的内存地址。 |
| size | 输入 | 图片大小,单位为Byte。 size大小需要和实际图片大小一致,否则可能会导致程序出现coredump情况。 |
| pictureType | 输入 | 图片类型,为下列三者之一: - PIXEL_FORMAT_ANY = 0 - PIXEL_FORMAT_JPEG = 1 - PIXEL_FORMAT_PNG = 2 |
父主题: 通用数据结构
DvppImageOutput
功能
DVPP图像输出定义。
结构定义
struct DvppImageOutput {
uint32_t width;
uint32_t height;
int32_t components;
uint32_t widthStride;
uint32_t heightStride;
uint32_t outImgDatasize;
}
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| width | 输出 | 输出图像的宽度。 |
| height | 输出 | 输出图像的高度。 |
| components | 输出 | 输出图像的通道个数。 |
| widthStride | 输出 | 输出对齐后的图像宽。 |
| heightStride | 输出 | 输出对齐后的图像高。 |
| outImgDatasize | 输出 | 输出图像的内存大小。 |
父主题: 通用数据结构
在线提单