shuffleNetv2 测试

代码地址：

https://github.com/yangyuke001/FIIQA_pytorch/blob/master/shufflenetv2.py

在tx2上， 这个速度90ms，尺度2是115ms，精度比原来高

 input = torch.randn(1, 3, 224, 224).cuda()
 

这个返回最后一层的特征：

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import torch.nn.functional as F
     
    
   

    

     

    
    

     

      from torch.autograd import Variable
     
    
   

    

     

    
    

     

      from collections import OrderedDict
     
    
   

    

     

    
    

     

      from torch.nn import init
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def conv_bn(inp, oup, stride):
     
    
   

    

     

    
    

     
 return nn.Sequential(
     
    
   

    

     

    
    

     

       nn.Conv2d(inp, oup, 3, stride, 1, bias=False),
     
    
   

    

     

    
    

     

       nn.BatchNorm2d(oup),
     
    
   

    

     

    
    

     

       nn.ReLU(inplace=True)
     
    
   

    

     

    
    

     

       )
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def conv_1x1_bn(inp, oup):
     
    
   

    

     

    
    

     
 return nn.Sequential(
     
    
   

    

     

    
    

     

       nn.Conv2d(inp, oup, 1, 1, 0, bias=False),
     
    
   

    

     

    
    

     

       nn.BatchNorm2d(oup),
     
    
   

    

     

    
    

     

       nn.ReLU(inplace=True)
     
    
   

    

     

    
    

     

       )
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def channel
     
    
  
 

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

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