Pytorch
whatsuo
这个作者很懒,什么都没留下…
展开
-
TorchScript官网教程
TorchScript是一种利用PyTorch代码创建可序列化和可优化模型的方法。 用TorchScript编写的任何代码都可以从Python进程中保存,然后可以在没有Python相关依赖的工程中被加载。 我们提供的工具可以将模型从纯Python程序逐步转换为可以独立于Python的TorchScript程序,例如,在C ++程序中。 用户可以使用熟悉的工具在PyTorch中训练模型,然后将模型导出到生产环境(生产环境中,由于性能和多线程原因,不能用Python程序来加载运行模型)。原创 2019-02-22 13:40:01 · 9948 阅读 · 1 评论 -
c++调用ECO-pytorch模型做行为识别
数据集:UCF101算法模型:ECO-pytorch,自己在UCF101做的finetune,top1准确率为75%左右,然后通过pytorch Script 将模型转为c++可调用的版本,后续有需求我可以上传这个模型。需要的库:opencv(网上很多教程,不再多说)和libtorch(官网可下)c++调用代码#include <torch/script.h> // One-s...原创 2019-02-26 19:18:41 · 1517 阅读 · 19 评论