skipnet_face

  

   

    

     

    
    

     

      """
     
    
   

    

     

    
    

     

      based on MobileNet implementation from
     
    
   

    

     

    
    

     

      https://github.com/rwightman/pytorch-image-models
     
    
   

    

     

    
    

     

      """
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import math
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

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

    

     

    
    

     
    """
     
    
   

    

     

    
    

     

       Ensures that all layers have a channel number that is divisible by 8
     
    
   

    

     

    
    

     

       function is taken from the original tf repo:
     
    
   

    

     

    
    

     

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

    

     

    
    

     

       """
     
    
   

    

     

    
    

     
    if min_value is None:
     
    
   

    

     

    
    

     

              min_value = divisor
     
    
   

    

     

    
    

     

          new_v = max(min_value, int(v + divisor / 2) // 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 h_sigmoid(nn.Module):
     
    
   

    

     

    
    

     
    def __init__(self, inplace=True):
     
    
   

    

     

    
    

     
        super(h_sigmoid, self).__init__()
     
    
   

    

     

    
    

     

              self.relu = nn.ReLU6(inplace=inplace)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    def forwa
     
    
  
 

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

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