opencv調用pytorch訓練好的模型

根據官方文檔知 cv2.dnn.readNetFromTorch() 中使用 torch.save() 方法保存的文件。 加載文件必須包含帶有導入網絡的序列化 nn.Module 對象，嘗試從序列化數據中消除自定義對象，以避免導入錯誤。 支持的圖層列表（即從Torch nn.Module類派生的對象實例）： nn.Sequential nn.Parallel nn.Concat nn.Linear

>>阅读原文<<