opencv调用pytorch训练好的模型

根据官方文档知 cv2.dnn.readNetFromTorch() 中使用 torch.save() 方法保存的文件。 加载文件必须包含带有导入网络的序列化 nn.Module 对象,尝试从序列化数据中消除自定义对象,以避免导入错误。 支持的图层列表(即从Torch nn.Module类派生的对象实例): nn.Sequential nn.Parallel nn.Concat nn.Linear
相关文章
相关标签/搜索