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

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