ShuffleNetV2(torch.jit.ScriptModule)

  

   

    

     

    
    

     

      #coding=utf-8
     
    
   

    

     

    
    

     

      from collections import OrderedDict
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import torch.nn.functional as F
     
    
   

    

     

    
    

     

      from torch.nn import init
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def _make_divisible(v, divisor, min_value=None):
     
    
   

    

     

    
    

     
 """
     
    
   

    

     

    
    

     

       This function is taken from the original tf repo.
     
    
   

    

     

    
    

     

       It ensures that all layers have a channel number that is divisible by 8
     
    
   

    

     

    
    

     

       It can be seen here:
     
    
   

    

     

    
    

     

       https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet.py
     
    
   

    

     

    
    

     

       :param v:
     
    
   

    

     

    
    

     

       :param divisor:
     
    
   

    

     

    
    

     

       :param min_value:
     
    
   

    

     

    
    

     

       :return:
     
    
   

    

     

    
    

     

       """
     
    
   

    

     

    
    

     
 if min_value is None:
     
    
   

    

     

    
    

     

       min_value = divisor
     
    
   

    

     

    
    

     

       new_v = max(min_value, int(v + divisor / 2.0) // divisor * divisor)
     
    
   

    

     

    
    

     
 # Make sure that round down does not go down by more than 10%.
     
    
   

    

     

    
    

     
 if new_v < 0.9 * v:
     
    
   

    

     

    
    

     

       new_v += divisor
     
    
   

    

     

    
    

     
 return new_v
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class SELayer(nn.Module):
     
    
   

    

     

    
    

     
 def __init__(self, channel, reduction=16):
     
    
  
 

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

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