faceboxes笔记

https://github.com/zisianw/FaceBoxes.PyTorch/blob/master/models/faceboxes.py

开始的maxpooling不能去掉，去掉就变慢好几倍

inception和reitnaface很像

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import torch.nn.functional as F
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class BasicConv2d(nn.Module):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 def __init__(self, in_channels, out_channels, **kwargs):
     
    
   

    

     

    
    

     

       super(BasicConv2d, self).__init__()
     
    
   

    

     

    
    

     

       self.conv = nn.Conv2d(in_channels, out_channels, bias=False, **kwargs)
     
    
   

    

     

    
    

     

       self.bn = nn.BatchNorm2d(out_channels, eps=1e-5)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 def forward(self, x):
     
    
   

    

     

    
    

     

       x = self.conv(x)
     
    
   

    

     

    
    

     

       x = self.bn(x)
     
    
   

    

     

    
    

     
 return F.relu(x, inplace=True)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class Inception(nn.Module):
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
  def __init__(self):
     
    
   

    

     

    
    

     

       super(Inception, self).__init__()
     
    
   

    

     

    
    

     

       self.branch1x1 = BasicConv2d(128, 32, kernel_size=1, padding=0)
     
    
   

    

     

    
    

     

       self.branch1x1_2 = BasicConv2d(128, 32, kern
     
    
  
 

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

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