libtorch调用模型

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

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> labels;
     
    
   

    

     

    
    

     
  if (LoadImageNetLabel(argv[2], labels)) {
     
    
   

    

     

    
    

     
 std::cout << "== Label loaded! Let's try it\n";
     
    
   

    

     

    
    

     

        } else {
     
    
   

    

     

    
    

     
 std::cerr &
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/93321009