示例代码
基本原理
- 调用acl.media.dvpp_create_channel接口创建图片数据处理的通道、调用acl.media.dvpp_destroy_channel接口销毁图片数据处理的通道。
- 调用acl.media.dvpp_jpeg_encode_async异步接口,将YUV格式图片编码成.jpg图片。
对于异步接口,还需调用acl.rt.synchronize_stream接口阻塞程序运行,直到指定Stream中的所有任务都完成。
- 调用acl.media.dvpp_jpeg_predict_enc_size接口根据输入图片描述信息、图片编码配置数据预估图片编码后所需的输出内存的大小。
实际输出内存大小可能与调用acl.media.dvpp_jpeg_predict_enc_size接口预估的内存大小存在差异,如果用户需要获取编码后的实际输出内存大小,可通过acl.media.dvpp_jpeg_encode_async接口的出参size获取。
- 输入、输出相关的约束要求,请参见功能及约束说明。
示例代码
您可以从获取样例中获取完整样例代码。
调用接口后,需增加异常处理的分支,并记录报错日志、提示日志,此处不一一列举。以下是关键步骤的代码示例,不可以直接拷贝运行,仅供参考。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
import acl # ...... # 1.pyACL初始化。 ret = acl.init() # 2.运行管理资源申请,包括Device、Context、Stream。 self.device_id = 0 ret = acl.rt.set_device(self.device_id) self.context, ret = acl.rt.create_context(&context_, self.device_id) self.stream, ret = acl.rt.create_stream(&stream_) # 3.创建图片数据处理通道时的通道描述信息,self.dvpp_channel_desc是acldvppChannelDesc类型。 self.dvpp_channel_desc = acl.media.dvpp_create_channel_desc() # 4.创建图片数据处理的通道。 ret = acl.media.dvpp_create_channel(self.dvpp_channel_desc) # 5.申请内存。 # 5.1 输入内存。 # 加载 YUV格式的图片数据,将通过acl.rt.memcpy接口将Host的图片数据传输到Device。 np_jpg = np.fromfile(self.path, dtype=np.byte) self.in_buffer_size = np_jpg.itemsize * np_jpg.size bytes_data = np_jpg.tobytes() np_jpg_ptr = acl.util.bytes_to_ptr(bytes_data) self.in_buffer_dev, ret = acl.media.dvpp_malloc(self.in_buffer_size) ret = acl.rt.memcpy(self.in_buffer_dev, self.in_buffer_size, np_jpg_ptr, self.in_buffer_size, 1) # 5.2 输出内存,按16对齐方式计算并申请Device内存self.out_buffer_dev,存放编码后的输出数据。 height_stride = ((h + 1) // 2) * 2 width_stride = ((w + 15) // 16) * 16 # malloc for output self.out_buffer_size = (width_stride * height_stride * 3) // 2 self.out_buffer_dev, ret = acl.media.dvpp_malloc(self.out_buffer_size # 6. 创建编码输入图片的描述信息,并设置各属性值。 # jpeg 编码设置输入yuv图片数据及size到图片描述。 self.pic_desc = acl.media.dvpp_create_pic_desc() ret = acl.media.dvpp_set_pic_desc_data(self.pic_desc, self.in_buffer_dev) ret = acl.media.dvpp_set_pic_desc_size(self.pic_desc, self.in_buffer_size) ret = acl.media.dvpp_set_pic_desc_format( self.pic_desc, PIXEL_FORMAT_YUV_SEMIPLANAR_420) ret = acl.media.dvpp_set_pic_desc_width(self.pic_desc, w) ret = acl.media.dvpp_set_pic_desc_height(self.pic_desc, h) ret = acl.media.dvpp_set_pic_desc_width_stride(self.pic_desc, width_stride) ret = acl.media.dvpp_set_pic_desc_height_stride(self.pic_desc, height_stride) # 7. 创建图片编码配置数据,设置编码质量, 预测输出的图片尺寸并分配内存。 # 编码质量范围[0, 100],其中level 0编码质量与level 100相近,而在[1, 100]内数值越小输出图片质量越差。 self.jpege_config = acl.media.dvpp_create_jpege_config() ret = acl.media.dvpp_set_jpege_config_level(self.jpege_config, 100) self.out_buffer_size, ret = acl.media.dvpp_jpeg_predict_enc_size( self.pic_desc, self.jpege_config) self.out_buffer_dev, ret = acl.media.dvpp_malloc(self.out_buffer_size) # 8. 执行异步编码,再调用acl.rt.synchronize_stream接口阻塞Host运行,直到指定Stream中的所有任务都完成。 acl.rt.set_context(self.context) np_out_size = np.array([self.out_buffer_size], dtype=np.int32) bytes_data = np_out_size.tobytes() np_out_size_ptr = acl.util.bytes_to_ptr(bytes_data) ret = acl.media.dvpp_jpeg_encode_async(self.dvpp_channel_desc, self.pic_desc, self.out_buffer_dev, np_out_size_ptr, self.jpege_config, self.stream) ret = acl.rt.synchronize_stream(self.stream) # 9.申请Host内存hostPtr。 # 9.1 将编码后的输出图片回传到Host,再将Host内存中的数据写入文件,写完文件后,需及时调用acl.rt.free_host接口释放Host内存。 # 9.2 创建numpy对象并转成host侧指针存储输出内容。 np_output = np.zeros(size, dtype=np.byte) bytes_data = np_output.tobytes() np_output_ptr = acl.util.bytes_to_ptr(bytes_data) size = int(np_out_size[0]) ret = acl.rt.memcpy(np_output_ptr, size, self.out_buffer_dev, size, 2) # 10. 编码结束后,释放资源,包括编码输入/输出图片的描述信息、编码输入/输出内存、通道描述信息、通道等。 ret = acl.media.dvpp_destroy_jpege_config(self.jpege_config) ret = acl.media.dvpp_destroy_pic_desc(self.pic_desc) ret = acl.media.dvpp_free(self.in_buffer_dev) ret = acl.media.dvpp_free(self.out_buffer_dev) ret = acl.media.dvpp_destroy_channel(self.dvpp_channel_desc) ret = acl.media.dvpp_destroy_channel_desc(self.dvpp_channel_desc) # 11. 释放运行管理资源。 ret = acl.rt.destroy_stream(self.stream) ret = acl.rt.destroy_context(self.context) ret = acl.rt.reset_device(self.device_id); ret = acl.finalize() # 12.pyACL去初始化。 # .... |
父主题: JPEGE图片编码