jkjung-avt / tensorrt_demos

TensorRT MODNet, YOLOv4, YOLOv3, SSD, MTCNN, and GoogLeNet
https://jkjung-avt.github.io/
MIT License
1.74k stars 545 forks source link

ModuleNotFoundError: No module named 'google.protobuf' #575

Closed Akhilesh1004 closed 2 years ago

Akhilesh1004 commented 2 years ago

Hi, I have some problem when i try "python3 yolo_to_onnx.py -m yolov4-tiny-416" it shows me "ModuleNotFoundError: No module named 'google.protobuf'" here is the full information

Traceback (most recent call last): File "yolo_to_onnx.py", line 58, in import onnx File "/usr/local/lib/python3.6/dist-packages/onnx/init.py", line 11, in from onnx.external_data_helper import load_external_data_for_model, write_external_data_tensors, convert_model_to_external_data File "/usr/local/lib/python3.6/dist-packages/onnx/external_data_helper.py", line 14, in from .onnx_pb import TensorProto, ModelProto File "/usr/local/lib/python3.6/dist-packages/onnx/onnx_pb.py", line 8, in from .onnx_ml_pb2 import * # noqa File "/usr/local/lib/python3.6/dist-packages/onnx/onnx_ml_pb2.py", line 7, in from google.protobuf.internal import enum_type_wrapper ModuleNotFoundError: No module named 'google.protobuf'

is it mean I did something wrong? but I am following the step to do it is anyone else have the same problem?