將 Caffe 權重轉換為 Keras 權重時出錯

Question

我正在拼命尋找一種將 Caffe 預訓練權重轉換為 Keras 權重的方法。 我正在使用caffe_weight_converter，但我一直收到下面的錯誤消息，我不知道為什么。 我的筆記本可以在這里找到

Traceback (most recent call last):
  File "caffe_weight_converter.py", line 24, in <module>
    import caffe
  File "/usr/local/lib/python3.6/dist-packages/caffe/python/caffe/__init__.py", line 1, in <module>
    from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
  File "/usr/local/lib/python3.6/dist-packages/caffe/python/caffe/pycaffe.py", line 13, in <module>
    from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \
ModuleNotFoundError: No module named 'caffe._caffe'

注意：為什么我使用 Python 3.6 版？ 因為如果我使用 3.7 版，我將收到另一條錯誤消息，說'caffe' module has no attribute Net 。 我看到有人使用 Python 3.6 而不是 Python 3.7 修復了錯誤，所以我決定試一試

Answer 1

確保將 caffe/python/caffe/ 中的文件夾復制到站點包文件夾中。 將它添加到您的路徑變量也是一個好主意。