libtorch调用模型

最新推荐文章于 2025-07-01 15:52:05 发布

AI算法网奇

最新推荐文章于 2025-07-01 15:52:05 发布

阅读量5.4k

点赞数

CC 4.0 BY-SA版权

分类专栏： c++入门宝典 pytorch知识宝典

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/93321009

pytorch知识宝典同时被 2 个专栏收录

512 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

c++入门宝典

359 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客介绍了如何利用libtorch库在C++中加载和执行预训练的PyTorch模型，通过给出的代码示例展示了读取图片并进行预测的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个也是调用模型，读取图片：

https://github.com/zacario-li/libtorch_cpp_mobilenetv2_5classes/blob/master/c%2B%2B/src/model_load.cpp

https://github.com/BIGBALLON/PyTorch-CPP/blob/master/prediction.cpp

 std::shared_ptr<torch::jit::script::Module> module =
      torch::jit::load(argv[1]);
  std::cout << "== Switch to GPU mode" << std::endl;
  // to GPU
  module->to(at::kCUDA);

  assert(module != nullptr);
  std::cout << "== ResNet50 loaded!\n";
  std::vector<std::string

了解本专栏