RuntimeError: Exporting the operator prim_DictConstruct to ONNX opset version 11 is not supported.

迷路在代码中

已于 2022-07-05 14:57:34 修改

阅读量1.1k

点赞数 1

分类专栏：开发工具 python linux 文章标签： pytorch 人工智能 python

于 2022-07-01 15:12:30 首次发布

本文链接：https://blog.csdn.net/chenran187906/article/details/125558778

版权

python 同时被 3 个专栏收录

35 篇文章 0 订阅

订阅专栏

linux

14 篇文章 0 订阅

订阅专栏

开发工具

6 篇文章 0 订阅

订阅专栏

【踩坑】

RuntimeError: Exporting the operator prim_DictConstruct to ONNX opset version 11 is not supported. Please open a bug to request ONNX export support for the missing operator.

问题场景：因为模型是多输出，模型输出本来是字典格式，在pytorch 保存pth转onnx时，出现无法支持字典操作，转模型格式报出错误。

解决：直接把网络输出格式更改。

# 把需要输出的字典 value 转 列表格式

out_values_list = []
out_key_list = ['prob', 'prob_logit', 'prob_map','height_prob','height_prob_logit',
                'center_mask', 'visit_mask', 'center_idx', 'offset', 'edge_map',
                 'seg_map']
for out_s in out_key_list:
     out_values_list.append(out[out_s])
print('test forward out keys:',out.keys())
out = tuple(out_values_list)