调用acl.media.dvpp_vpc_crop_resize_paste_async实现抠图缩放贴图时,支持指定缩放算法。
调用接口后,需增加异常处理的分支,并记录报错日志、提示日志,此处不一一列举。以下是关键步骤的代码示例,不可以直接拷贝运行,仅供参考。
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 |
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(self.device_id) self.stream, ret = acl.rt.create_stream() # 3.创建图片数据处理通道时的通道描述信息,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.创建输出图片的描述信息,并设置各属性值。 self.output_desc = acl.media.dvpp_create_pic_desc() assert self.output_desc is not None out_buffer_size = self.set_picture_desc(self.output_desc, w // 2, h // 2, "output", 0) # 6.创建输入图片的描述信息,并设置各属性值。 self.input_desc = acl.media.dvpp_create_pic_desc() assert self.input_desc is not None self.set_picture_desc(self.input_desc, w, h, "input", 0) # 7.加载图片数据,并拷贝到Device侧。 np_yuv = np.fromfile(path, dtype=np.byte) in_buffer_size = np_yuv.itemsize * np_yuv.size bytes_data = np_yuv.tobytes() np_yuv_ptr = acl.util.bytes_to_ptr(bytes_data) ret = acl.rt.memcpy(self.dev_buffer["input_0"], in_buffer_size, np_yuv_ptr, in_buffer_size, ACL_MEMCPY_HOST_TO_DEVICE) assert ret == 0 # 8.指定抠图、贴图区域的位置。 self.crop_area = acl.media.dvpp_create_roi_config(w // 2, w - 1, h // 2, h - 1) self.paste_area = acl.media.dvpp_create_roi_config(w // 4, w // 2 - 1, h // 4, h // 2 - 1) # 9.执行异步抠图贴图缩放。 self.resize_config = acl.media.dvpp_create_resize_config() ret = acl.media.dvpp_vpc_crop_resize_paste_async(self.dvpp_channel_desc, self.input_desc, self.output_desc, self.crop_area, self.paste_area, self.resize_config, self.stream) assert ret == 0 ret = acl.rt.synchronize_stream(self.stream) assert ret == 0 # 10.将输出数据拷贝到Host侧。 np_output = np.zeros(out_buffer_size, dtype=np.byte) bytes_data = np_output.tobytes() np_output_ptr = acl.util.bytes_to_ptr(bytes_data) ret = acl.rt.memcpy(np_output_ptr, out_buffer_size, self.dev_buffer["output_0"], out_buffer_size, ACL_MEMCPY_DEVICE_TO_HOST) assert ret == 0 # 11.释放资源,包括输入/输出图片的描述信息、输入/输出内存、通道描述信息、通道等。 ret = acl.media.dvpp_destroy_pic_desc(self.input_desc) ret = acl.media.dvpp_destroy_pic_desc(self.output_desc) # dev_buffer是字典对象,存储device侧申请内存。 for key in self.dev_buffer.keys(): if self.dev_buffer[key]: ret = acl.media.dvpp_free(self.dev_buffer[key]) if self.dvpp_channel_desc: ret = acl.media.dvpp_destroy_channel(self.dvpp_channel_desc) assert ret == 0 ret = acl.media.dvpp_destroy_channel_desc(self.dvpp_channel_desc) assert ret == 0 # 12.释放运行管理资源。 ret = acl.rt.destroy_stream(self.stream) ret = acl.rt.destroy_context(self.context) ret = acl.rt.reset_device(self.device_id) # 13.pyACL去初始化。 ret = acl.finalize() # ...... |