pvanet 检测

tf的

https://github.com/lxyzler/EAST-PVANET

  

   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torchvision
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import torch.nn.functional as F
     
    
   

    

     

    
    

     

      from utils import *
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      # XXX: Not tested
     
    
   

    

     

    
    

     

      class CReLU(nn.Module):
     
    
   

    

     

    
    

     
 def __init__(self, act=F.relu):
     
    
   

    

     

    
    

     

       super(CReLU, self).__init__()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       self.act = act
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 def forward(self, x):
     
    
   

    

     

    
    

     

       x = torch.cat((x, -x), 1)
     
    
   

    

     

    
    

     

       x = self.act(x)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 return x
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class ConvBnAct(nn.Module):
     
    
   

    

     

    
    

     
 def __init__(self, n_in, n_out, **kwargs):
     
    
   

    

     

    
    

     

       super(ConvBnAct, self).__init__()
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       self.conv = nn.Conv2d(n_in, n_out, bias=False, **kwargs)
     
    
   

    

     

    
    

     

       self.bn = nn.BatchNorm2d(n_out)
     
    
   

    

     

    
    

     

       self.act = F.relu
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 def forward(self, x):
     
    
   

    

     

    
    

     

       x = self.conv(x)
     
    
   

    

     

    
    

     

       x = self.bn(x)
     
    
   

    

     

    
    

     

       x = self.act(x)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 return x
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class mCReLU_base(nn.Module):
     
    
   

    

     

    
    

     
 def __init__(self, n_i
     
    
  
 

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

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