I run it, and it generate errors

[antran89]

Thank you for writing the codes and publish it into the world.

python3 convertCaffe.py
Traceback (most recent call last):
  File "convertCaffe.py", line 5, in <module>
    import caffe
ModuleNotFoundError: No module named 'caffe'

I am not able to run the conversion. What is the version of caffe here? Is it the original BVLC Caffe. Thank you.

[Wulingtian]

https://github.com/Wulingtian/yolov5_caffe

[antran89]

Hi @Wulingtian, is it your repo working with Python2? I am having problems with compiling Caffe with Python3. I am thinking of using Python2.

[lwxGitHub123]

你好，这个问题解决了吗？我也遇到这个问题了。用python2装不了onnx