caffe 轉caffe2模型出現的問題

2021-08-09 14:56:26 字數 375 閱讀 6837

剛開始接觸caffe,用nvidia tx2跑了ssd模型,達到8~9幀。

把caffe模型轉caffe2模型時,轉到reshape層時出現錯誤

valueerror: unknown argument type: key=values value=[

..........

發現是protobuffer 版本問題,參考

解壓,編譯,安裝

typeerror

:__init__

()got an unexpected keyword argument

'syntax

'

原因是protoc和python-protobuf版本不一致

caffe2 安裝與介紹

一早發現caffe2的較成熟的release版發布了 the first production ready release 那麼深度學習平台在之後一段時間也是會出現其與tensorflow相互競爭的局面。從開啟這個caffe2的官網就會發現,有了facebook的支援,連介面也好看多了。不過再仔細看...

Caffe2的安裝詳述

caffe2的安裝過程區分ubuntu的不同版本,在ubuntu 14.04和 ubuntu 16.04下的相關命令有所不同,最重要的是caffe2主要支援python2.7版本,以下python和pip命令預設情況下都是指python2.7版本,經過筆者測試,caffe2版本暫時不支援python...

Caffe2 新增CUDA路徑

在使用eclipse時候,仍然出現 libcuda.so 找不到的情況.原因 cuda環境沒有設定到系統環境中。方案 設定cuda path 在bashrc中新增路徑,只對當前使用者產生影響。sudo gedit bashrc在最後行新增內容 export path usr local cuda b...