CANN/pyasc图像加载API

张开发
2026/5/9 17:29:37 15 分钟阅读

分享文章

CANN/pyasc图像加载API
asc.language.basic.load_image_to_local【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyascasc.language.basic.load_image_to_local(dst: LocalTensor, load_data_params: LoadImageToLocalParams) → None将图像数据从GM搬运到A1/B1。 搬运过程中可以完成图像预处理操作包括图像翻转改变图像尺寸抠图裁边缩放伸展以及色域转换类型转换等。 图像预处理的相关参数通过set_aipp_functions进行配置。对应的Ascend C函数原型template typename T __aicore__ inline void LoadImageToLocal(const LocalTensorT dst, const LoadImageToLocalParams loadDataParams)参数说明dst输出目的操作数类型为LocalTensor支持的TPosition为A1/B1。LocalTensor的起始地址需要保证32字节对齐。不同产品支持的数据类型不同Atlas A3 训练/推理系列产品int8_t/halfAtlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件int8_t/halfAtlas 200I/500 A2 推理产品uint8_t/int8_t/halfload_data_params输入LoadData参数结构体类型为LoadImageToLocalParams包含以下子参数horiz_size输入从源图中加载图片的水平宽度单位为像素取值范围horiz_size∈[2, 4095]。vert_size输入从源图中加载图片的垂直高度单位为像素取值范围vert_size∈[2, 4095]。horiz_start_pos输入加载图片在源图片上的水平起始地址单位为像素取值范围horiz_start_pos∈[0, 4095]默认为0。注意当输入图片为YUV420SP、XRGB8888、RGB888和YUV400格式时该参数需要是偶数。vert_start_pos输入加载图片在源图片上的垂直起始地址单位为像素取值范围vert_start_pos∈[0, 4095]默认为0。注意当输入图片为YUV420SP格式时该参数需要是偶数。src_horiz_size输入源图像水平宽度单位为像素取值范围src_horiz_size∈[2, 4095]。注意当输入图片为YUV420SP格式时该参数需要是偶数。top_pad_size输入目的图像顶部填充的像素数取值范围top_pad_size∈[0, 32]默认为0。进行数据填充时使用需要先调用SetAippFunctions(ISASI)通过AippPaddingParams配置填充的数值再通过topPadSize、botPadSize、leftPadSize、rightPadSize配置填充的大小范围。bot_pad_size输入目的图像底部填充的像素数取值范围bot_pad_size∈[0, 32]默认为0。left_pad_size输入目的图像左边填充的像素数取值范围left_pad_size∈[0, 32]默认为0。right_pad_size输入目的图像右边填充的像素数取值范围right_pad_size∈[0, 32]默认为0。sid输入预留参数为后续功能保留开发者暂时无需关注使用默认值即可。返回值说明无约束说明操作数地址对齐要求请参见 《Ascend C算子开发接口》 中的“通用说明和约束-通用地址对齐约束”。加载到dst的图片的大小加padding的大小必须小于等于L1的大小。对于XRGB输入格式的数据芯片在处理的时候会默认丢弃掉第四个通道的数据所以需要在set_aipp_functions接口里设置好通道交换的参数后输出RGB格式的数据。调用示例dst asc.LocalTensor(dtypeasc.float16, posasc.TPosition.A1, addr0, tile_size128) load_data_params asc.LoadImageToLocalParams(2, 2, 0, 0, 2, 0, 0, 0, 0) asc.load_image_to_local(dst, load_data_params)【免费下载链接】pyasc本项目为Python用户提供算子编程接口支持在昇腾AI处理器上加速计算接口与Ascend C一一对应并遵守Python原生语法。项目地址: https://gitcode.com/cann/pyasc创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

更多文章