??主頁:吾名招財
??簡介:工科學碩,研究方向機器視覺,愛好較廣泛…
???簽名:面朝大海,春暖花開!
多onnx模型合并導出調研(文本檢測+方向分類+文本識別)
- 引言
- 1,嘗試合并兩個模型(文本方向分類+文本識別模型)(并行合并)
- (1)文本方向分類
- (2)文本識別模型
- (3)onnx合并(并行)
- 2,將預處理加入到一個模型中
- (1)原始模型
- (2)預處理函數轉換成onnx模型
- (3)新的模型
- 3,onnx轉換成torch
- 4,相關結論
- 相關鏈接
引言
目標:
??部署端只用一個模型實現文本檢測及文本識別。
??目前是文本檢測、文本方向分類、文本識別三個模型分別訓練,然后部署的話需要三個分別部署并使用自定義流程連接,現在需要想將其三個模型按照順序合并成為一個模型,然后再用這一個模型部署(方便部署),此時僅僅是簡化了部署流程。
??一種是上面的,另一種是端到端的算法(PGNET),這是一個算法,這個算法同時進行文本檢測及文本識別輸出,最終一個模型訓練輸出,因后續要對文本識別通用模型進行部署,故此端到端方法暫時舍棄。
模型拼接調研驗證:
1,直接對onnx模型進行操作合并拼接測試驗證。