您也可以调用自己的后处理函数。以 demo_yolov3.cpp 和 demo_classification.cpp 文件为例。使用 vitis::ai::DpuTask::create
或 vitis::ai::DpuRunner::create_dpu_runner 创建任务,完成 DPU 处理后,即可调用后处理函数。下图中的 post_process
函数显示了用户后处理代码的示例。
图 1. 用户后处理代码示例
您也可以调用自己的后处理函数。以 demo_yolov3.cpp 和 demo_classification.cpp 文件为例。使用 vitis::ai::DpuTask::create
或 vitis::ai::DpuRunner::create_dpu_runner 创建任务,完成 DPU 处理后,即可调用后处理函数。下图中的 post_process
函数显示了用户后处理代码的示例。