# 自己的模型为: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
opencv-C++调用torch-python训练好的网络
最新推荐文章于 2024-01-10 15:24:45 发布