opencv-C++调用torch-python训练好的网络

#  自己的模型为:model
traced_script_module = torch.jit.trace(model, src_inp)
traced_script_module.save("model.pt")

#  安装libtorch

#  使用pytorch训练模型时,如果模型里有BN或者Dropout层,测试之前需要加上 model = model.eval()

更完整的方法:https://zhuanlan.zhihu.com/p/191569603

你可能感兴趣的:(python,opencv,c++)