Skip to main content

TensorDesc

MakeBorderConfig​

功能​

用于在图像处理补边功能，定义具体的补边配置，可设置图像补边的左、右、上、下像素个数、补边颜色常数和补边类型。

结构定义​

struct MakeBorderConfig {
    enum BorderType {
        BORDER_CONSTANT = 0,
        BORDER_REPLICATE,
        BORDER_REFLECT,
        BORDER_REFLECT_101
    };
    uint32_t left;
    uint32_t right;
    uint32_t top;
    uint32_t bottom;
    uint32_t channel_zero;
    uint32_t channel_one;
    uint32_t channel_two;
    BorderType borderType;
};

参数说明​

参数名输入/输出说明
left输入左侧补边像素数量。
right输入右侧补边像素数量。
top输入上方补边像素数量。
bottom输入下方补边像素数量。
channel_zero输入0号通道取值，范围[0, 255]。
channel_one输入1号通道取值，范围[0, 255]。
channel_two输入2号通道取值，范围[0, 255]。
borderType输入补边类型。
- BORDER_CONSTANT：添加有颜色的常数值边界。
- BORDER_REPLICATE：重复最后一个元素。 举例：aaaaaa|a*****h|hhhhhhh（其中*表示任意图像元素）。
- BORDER_REFLECT：边界元素的镜像，镜像包括边界元素。 举例：ba|abc*******fgh|hg（其中*表示任意图像元素）。
- BORDER_REFLECT_101：边界元素的镜像，镜像不包括边界元素。 举例：cb|abc****fgh|gf（其中*表示任意图像元素）。

:::note 说明 “channel_zero”、“channel_one”、“channel_two”，依次对应各图像通道，例如：使用RGB时，“channel_zero”对应R，“channel_one”对应G通道，“channel_two”对应B通道。 :::

父主题： 通用数据结构

BaseTensor​

功能​

定义一个张量数据的结构。

结构定义​

struct BaseTensor { 
     void* buf; 
     std::vector<int> shape; 
     size_t size; 
 &#125;

参数说明​

参数名输入/输出说明
buf输入张量的数据。
shape输入张量的形状。
size输入张量数据对应内存大小，单位为Byte。 size大小应与实际内存大小一致，否则可能会导致程序出现coredump情况。

父主题： 通用数据结构

TensorDesc​

功能​

张量描述信息定义。

结构定义​

struct TensorDesc &#123; 
    size_t tensorSize; 
    std::string tensorName; 
    std::vector<int64_t> tensorDims; 
&#125;;

参数说明​

参数名输入/输出说明
tensorSize输出张量的大小，单位为Byte。
tensorName输出张量名称。
tensorDims输出张量维度。

父主题： 通用数据结构

ModelDesc​

功能​

模型描述信息定义。

结构定义​

struct ModelDesc &#123; 
    std::vector<TensorDesc> inputTensors; 
    std::vector<TensorDesc> outputTensors; 
    std::vector<size_t> batchSizes; 
    bool dynamicBatch; 
&#125;;

参数说明​

参数名输入/输出说明
inputTensors输出输入张量描述。
outputTensors输出输出张量描述。
batchSizes输出batch大小。
dynamicBatch输出是否为动态Batch。

父主题： 通用数据结构

ModelDataset​

功能​

模型数据集定义。

结构定义​

struct ModelDataset &#123; 
    void* mdlDataPtr; 
    size_t dynamicBatchSize; 
&#125;;

参数说明​

参数名输入/输出说明
mdlDataPtr输入模型数据集数据的内存地址。
dynamicBatchSize输入模型推理时的批量大小。

父主题： 通用数据结构

在线提单