Pytorch模型轉(zhuǎn)onnx模型實例
如下所示：
import io
import torch
import torch.onnx
from models.C3AEModel import PlainC3AENetCBAM
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
def test():
model = PlainC3AENetCBAM()
pthfile = r'/home/joy/Projects/models/emotion/PlainC3AENet.pth'
loaded_model = torch.load(pthfile, map_location='cpu')
# try:
# loaded_model.eval()
# except AttributeError as error:
# print(error)
model.load_state_dict(loaded_model['state_dict'])
# model = model.to(device)
#data type nchw
dummy_input1 = torch.randn(1, 3, 64, 64)
# dummy_input2 = torch.randn(1, 3, 64, 64)
# dummy_input3 = torch.randn(1, 3, 64, 64)
input_names = [ "actual_input_1"]
output_names = [ "output1" ]
# torch.onnx.export(model, (dummy_input1, dummy_input2, dummy_input3), "C3AE.onnx", verbose=True, input_names=input_names, output_names=output_names)
torch.onnx.export(model, dummy_input1, "C3AE_emotion.onnx", verbose=True, input_names=input_names, output_names=output_names)
if __name__ == "__main__":
test()
直接將PlainC3AENetCBAM替換成需要轉(zhuǎn)換的模型，然后修改pthfile，輸入和onnx模型名字然后執(zhí)行即可。
注意：上面代碼中注釋的dummy_input2，dummy_input3，torch.onnx.export對應(yīng)的是多個輸入的例子。
在轉(zhuǎn)換過程中遇到的問題匯總
RuntimeError: Failed to export an ONNX attribute, since it's not constant, please try to make things (e.g., kernel size) static if possible
在轉(zhuǎn)換過程中遇到RuntimeError: Failed to export an ONNX attribute, since it's not constant, please try to make things (e.g., kernel size) static if possible的錯誤。
根據(jù)報的錯誤日志信息打開/home/joy/.tensorflow/venv/lib/python3.6/site-packages/torch/onnx/symbolic_helper.py，在相應(yīng)位置添加print之后，可以定位到具體哪個op出問題。
例如：
在相應(yīng)位置添加
print（v.node()）
輸出信息如下：
%124 : Long() = onnx::Gather[axis=0](%122, %121), scope: PlainC3AENetCBAM/Bottleneck[cbam]/CBAM[cbam]/ChannelGate[ChannelGate] # /home/joy/Projects/models/emotion/WhatsTheemotion/models/cbam.py:46:0
原因是pytorch中的tensor.size(1)方式onnx識別不了，需要修改成常量。
以上這篇Pytorch模型轉(zhuǎn)onnx模型實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python實現(xiàn)爬蟲爬取NBA數(shù)據(jù)功能示例
這篇文章主要介紹了Python實現(xiàn)爬蟲爬取NBA數(shù)據(jù)功能,涉及Python針對URL模塊、字符串、列表遍歷、Excel寫入等相關(guān)操作技巧,需要的朋友可以參考下2018-05-05
這篇文章主要介紹了Python用Bottle輕量級框架進行Web開發(fā)的相關(guān)資料,需要的朋友可以參考下2016-06-06
這篇文章主要介紹了Python實現(xiàn)像awk一樣分割字符串,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-09-09
本講再次提及變量和引用對象，就是要讓看官對變量和賦值有一個知其然和知其所以然的認識。當(dāng)然，最后能不能達到此目的，主要看我是不是說的通俗易懂了。如果您沒有明白，就說明我說的還不夠好，可以聯(lián)系我，我再為您效勞。2014-09-09
python3中No module named _ssl的問題解決
本文主要介紹了python3中No module named _ssl的問題解決,這個錯誤表示Python導(dǎo)入_ssl模塊時失敗,文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2024-08-08
這篇文章主要介紹了python的exec、eval使用分析，具有一定借鑒價值，需要的朋友可以參考下。2017-12-12
anaconda3:conda not found報錯問題解決
這篇文章主要給大家介紹了關(guān)于anaconda3:conda not found報錯問題解決的相關(guān)資料,Anaconda指的是一個開源的Python發(fā)行版本,其包含了conda、Python等180多個科學(xué)包及其依賴項,需要的朋友可以參考下2023-10-10
詳細解讀Python的web.py框架下的application.py模塊
這篇文章主要介紹了Python的web.py框架下的application.py模塊,作者深入分析了web.py的源碼,需要的朋友可以參考下2015-05-05
Jupyter Notebook切換conda虛擬環(huán)境的實現(xiàn)步驟
本文主要介紹了Jupyter Notebook切換conda虛擬環(huán)境的實現(xiàn)步驟，文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-07-07
python GUI庫圖形界面開發(fā)之PyQt5中QWebEngineView內(nèi)嵌網(wǎng)頁與Python的數(shù)據(jù)交互傳參詳細方法
這篇文章主要介紹了python GUI庫圖形界面開發(fā)之PyQt中QWebEngineView內(nèi)嵌網(wǎng)頁與Python的數(shù)據(jù)交互詳細方法實例,需要的朋友可以參考下2020-02-02