libtorch 调用pytorch模型

我测试的速度没有提升

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      int main(int argc, const char* argv[])
     
    
   

    

     

    
    

     

      {
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
	//size_t len = url.length();//获取字符串长度
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
	char sBuf[1024];
     
    
   

    

     

    
    

     
	char *ptr;
     
    
   

    

     

    
    

     
	if (GetModuleFileNameA(NULL, sBuf, sizeof(sBuf)))
     
    
   

    

     

    
    

     

      	{
     
    
   

    

     

    
    

     

      		ptr = strrchr(sBuf, '\\');
     
    
   

    

     

    
    

     
		if (ptr)
     
    
   

    

     

    
    

     

      			*ptr = '\0';
     
    
   

    

     

    
    

     

      		SetCurrentDirectoryA(sBuf);
     
    
   

    

     

    
    

     

      	}
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      	torch::DeviceType device_type;
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
	if (torch::cuda::is_available()) {
     
    
   

    

     

    
    

     

      		device_type = torch::kCUDA;
     
    
   

    

     

    
    

     

      	}
     
    
   

    

     

    
    

     
	else {
     
    
   

    

     

    
    

     

      		device_type = torch::kCPU;
     
    
   

    

     

    
    

     

      	}
     
    
   

    

     

    
    

     
	torch::Device device(device_type);
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      try
     
    
   

    

     

    
    

     

      	{
     
    
   

    

     

    
    

     
		std::shared_ptr<torch::jit::script::Module> module = torch::jit::load("../models/yolo_model.pt");
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      		assert(module != nullptr);
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
		//module->to(at::kCUDA);
     
    
   

    

     

    
    

     
		module->to(device);
     
    
   

    

     

    
    

     

      		torch::Tensor img_tensor = torch::rand({ 1, 3,352,352 }).to(device);
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
		//img_tensor.to(at::kCUDA)
     
    
   

    

     

    
    

     
		//auto img_var = torch::autograd::make_variable(img_tensor, false).to(device);
     
    
   

    

     

    
    

     
		for (int i = 0; i < 20;i++) {
     
    
   

    

     

    
    

     

      			DWORD st
     
    
  
 

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

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