hon 调用 QNN 做批量图像推理,频繁出现 GIL 锁阻塞导致推理吞吐量上不去,如何解决?

风声wind 2026-08-06 11:56:51

hon 调用 QNN 做批量图像推理,频繁出现 GIL 锁阻塞导致推理吞吐量上不去,如何解决?
 

...全文
28 1 打赏 收藏 转发到动态 举报
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
老杨261 08-06 20:40
  • 打赏
  • 举报
回复

阻塞原因
Python 全局 GIL 锁使得同一时刻仅有一个线程执行 CPU 运算,即便 NPU 空闲,多线程推理预处理阶段也会被 GIL 限制,无法并发送帧至 NPU。
解决方案
将图像预处理逻辑剥离 Python,封装为独立 C++ 动态库,脱离 GIL 运行;
使用多进程架构替代多线程,每个进程单独持有一套 QNN 模型上下文,进程间通过队列传递图像路径;
NPU 设置批量推理 batch 参数,单次送入多张图片推理,减少推理调用次数。
多进程调度 Python 示例
python
运行
from multiprocessing import Pool, Queue

def infer_subprocess(img_queue):
# 子进程单独初始化QNN模型,互不干扰
from qnn_wrapper import QnnModel
model = QnnModel("batch_model.dlc")
while True:
img_path = img_queue.get()
if img_path is None: break
img = cv2.imread(img_path)
res = model.infer(img)

if name == "main":
task_queue = Queue()
process_num = 2
pool = Pool(processes=process_num, initializer=infer_subprocess, initargs=(task_queue,))
# 投递图片任务
for path in img_list:
task_queue.put(path)
for _ in range(process_num):
task_queue.put(None)
pool.close()
pool.join()
优化效果
规避 GIL 限制后,骁龙 AI PC 批量图片推理吞吐量可提升 2 倍以上。

7,651

社区成员

发帖
与我相关
我的任务
社区描述
本论坛以AI、WoS 、XR、IoT、Auto、生成式AI等核心板块组成,为开发者提供便捷及高效的学习和交流平台。 高通开发者专区主页:https://qualcomm.csdn.net/
人工智能物联网机器学习 技术论坛(原bbs) 北京·东城区
社区管理员
  • csdnsqst0050
  • chipseeker
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧