最近，在把pytorch模型轉為onnx時，遇到乙個問題，有28%的資料對不齊，因為接觸的是顯著性task的，裡面用了一些upsample層。
查了pytorch官方文件後發現，這裡的upsample只支援nearest一種模式，而我用的是bilinear，在改變了這個之後，結果就對的齊了。
建議：先去官方文件看一下哪些運算元支援哪些運算元不支援，以及別用function函式，得用torch.nn裡面的層。
官方文件：
onnx模型轉tensorflow模型
onnx是開源神經網路交換平台，有了它基本上不用糾結用什麼深度學習框架的問題了。我現在記錄一下怎麼將onnx模型轉換成tensorflow模型。1 安裝tensorflow和onnx 我是通過anaconda安裝的。詳情 這個部落格記載了安裝anaconda和onnx的詳情，安裝好anaconda後...
pt模型訓練，轉onnx
pytorch訓練yolov5s.pt，之後轉onnx。以檢測條形碼為例，1分類問題，訓練yolov5s.pt。2 data資料夾裡，images放入樣本，labels放入對應的txt標籤 labelimg製作，注意選擇yolo模式 新建yaml檔案 上圖中的barcode.yaml 用於指定路徑和...
onnx模型轉tensorflow的pb格式
遇到了很多坑，最常見的就是少了什麼模組，或某個模組沒有啥屬性，如 module tensorflow has no attribute ceil 記錄一下現在轉換成功的方式，首先 git clone 之後在終端安裝 pip install e.pip install tensorflow addon...