opencv dnn完整示例

目录

python 导出onnx：

python预测例子：

c++ dnn调用例子：

python 导出onnx：

  

   

    

     

    
    

     

         weights = 'xxxx.pth'
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          model_cls= Model_net(model_path=weights, num_classes=2)
     
    
   

    

     

    
    

     

          state_dict = torch.load(weights)
     
    
   

    

     

    
    

     

          model_cls.load_state_dict(state_dict)
     
    
   

    

     

    
    

     

          model_cls.eval()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          dummy_input1 = torch.randn(1, 3, 64, 64)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          onnx_name = "smoke_skipnet.onnx"
     
    
   

    

     

    
    

     

          torch.on
     
    
  
 

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

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