解决ONNX export failure: cannot import name ‘builder‘ from ‘google.protobuf.internal‘ 问题

Akaset

于 2022-10-05 20:33:01 发布

阅读量5.9k

点赞数 9

CC 4.0 BY-SA版权

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_61328371/article/details/127176489

本文记录了在将YOLOv5-lite的.pt文件转换为.onnx格式过程中遇到的问题及解决方法。主要问题是ONNX导出失败，无法导入protobuf的builder模块。通过更新protobuf到最新版本解决了该问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在进行训练完YOLOv5-lite后，想把pt文件转换成onnx文件的时候报错

报错如下：


Fusing layers... 
F:\Anaconda3\envs\yolov5\lib\site-packages\torch\functional.py:478: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered i
nternally at  C:\cb\pytorch_1000000000000\work\aten\src\ATen\native\TensorShape.cpp:2895.)
  return _VF.meshgrid(tensors, **kwargs)  # type: ignore[attr-defined]
Model Summary: 167 layers, 713111 parameters, 0 gradients, 2.5 GFLOPS
  (m): ModuleList(
    (1): Conv2d(96, 21, kernel_size=(1, 1), stride=(1, 1))
  )
)
ONNX export failure: cannot import name 'builder' from 'google.protobuf.internal' (F:\Anaconda3\envs\yolov5\lib\site-packages\google\protobuf\internal\__init__.py)

查阅sackoverflew后，得到如下解决办法

即更新protobuf至最新版即可

 pip install --upgrade protobuf

有效嘻嘻