retinaface box添加layyer

  

   

    

     

    
    

     

      import argparse
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import torch.nn.functional as F
     
    
   

    

     

    
    

     

      # from torchvision.models._utils import IntermediateLayerGetter
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      from Pelee_network_v2 import PeleeNet
     
    
   

    

     

    
    

     

      from Snet import Snet
     
    
   

    

     

    
    

     

      # from bi_fpn import BIFPN
     
    
   

    

     

    
    

     

      from bifpn import BiFPN
     
    
   

    

     

    
    

     

      from darknet import DarkNet
     
    
   

    

     

    
    

     

      from data import cfg_mnet, cfg_shuffle
     
    
   

    

     

    
    

     

      from models.net import MobileNetV1 as MobileNetV1
     
    
   

    

     

    
    

     

      from models.net import FPN as FPN
     
    
   

    

     

    
    

     

      from models.net import SSH as SSH
     
    
   

    

     

    
    

     

      # from models.peleenet import PeleeNet
     
    
   

    

     

    
    

     

      from models.shufflenet_yolo import ShuffleNetV2
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class ClassHead(nn.Module):
     
    
   

    

     

    
    

     
    def __init__(self,inchannels=512,num_anchors=3,num_classes=2):
     
    
   

    

     

    
    

     
        super(ClassHead,self).__init__()
     
    
   

    

     

    
    

     

              self.num_anchors = num_anchors
     
    
   

    

     

    
    

     

              self.num_classes=num_classes
     
    
   

    

     

    
    

     

              self.conv1x1 = nn.Conv2d(inchannels,self.num_anchors*2,kernel_size=(1,1),stride=1,padding=0)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    def forward(self,x):
     
    
   

    

     

    
    

     

              out = self.
     
    
  
 

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

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