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中的这些类的某些等效项也可以成功导入。