opencv調用pytorch訓練好的模型

根據官方文檔知 cv2.dnn.readNetFromTorch() 中使用 torch.save() 方法保存的文件。

加載文件必須包含帶有導入網絡的序列化 nn.Module 對象，嘗試從序列化數據中消除自定義對象，以避免導入錯誤。

支持的圖層列表（即從Torch nn.Module類派生的對象實例）：

• nn.Sequential
• nn.Parallel
• nn.Concat
• nn.Linear
• nn.SpatialConvolution
• nn.spatialMaxPooling, nn.SpatialAveragePooling
• nn.ReLU, nn.TanH, nn.Sigmoid
• nn.Reshape
• nn.SoftMax, nn.LogSoftMax

cunn，cudnn和fbcunn中的這些類的某些等效項也可以成功導入。