可以看出用C++ libtorch加载模型预测的结果跟pytorch是一致的 代码 csdn:download.csdn.net/downl github:github.com/tashaxing/li 支持是知识分享的动力,有问题可扫码哦编辑于 2020-07-13 20:13 机器学习 PyTorch C++ 赞同12 条评论 分享喜欢收藏申请转载 ...
数据集部分都只做了简单处理,所以其实还有很多的优化空间。相关代码及训练数据及已上传至github,有疑问的同学请题Issue或在博客下方留言。