V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
V2EX 提问指南
suifengingo
V2EX  ›  问与答

如何把基于深度学习的图像分类模型做成 LibTorch 调用的动态连接库?

  •  
  •   suifengingo · 2021-02-23 22:40:07 +08:00 · 884 次点击
    这是一个创建于 1356 天前的主题,其中的信息可能已经有所发展或是发生改变。
    向大佬请教一个问题,基于深度学习图像分类模型训练完成后,如何将其做成 LibTorch 调用的动态连接库,有没有那种教程可供参考呢?本人小白,望多多指教,虚心请教!
    2 条回复    2021-02-24 14:13:22 +08:00
    suifengingo
        1
    suifengingo  
    OP
       2021-02-24 11:27:46 +08:00
    虚心请教 ing
    sangleft
        2
    sangleft  
       2021-02-24 14:13:22 +08:00
    没搞懂你的问题,保存的模型文件通过 libtorch 调用就好,编译成动态库的话我是用 C++包装了一下推理函数然后做成 dll 。而且 C++项目中也要有模型结构,还挺麻烦。


    19 年做的,是要比 python 快一丢丢,现在可能有其他方式吧。
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   2736 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 29ms · UTC 15:06 · PVG 23:06 · LAX 07:06 · JFK 10:06
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.