PriorBox 改进

不改中心，把边框改为非对称的：

self.clip=True即可

超过边界，锚框移动到边界上：

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      from itertools import product as product
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      from math import ceil
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      class PriorBox(object):
     
    
   

    

     

    
    

     
    def __init__(self, cfg, image_size=None, x_scale=1, y_scale=1):
     
    
   

    

     

    
    

     
        super(PriorBox, self).__init__()
     
    
   

    

     

    
    

     

              self.min_sizes = cfg['min_sizes']
     
    
   

    

     

    
    

     

              self.steps = cfg['steps']
     
    
   

    

     

    
    

     

              self.clip = cfg['clip']
     
    
   

    

     

    
    

     

              self.image_size = image_size
     
    
   

    

     

    
    

     

              self.feature_maps = [[ceil(self.image_size[0]/step), ceil(self.image_size[1]/step)] for step in self.steps]
     
    
   

    

     

    
    

     

              self.name = "s"
     
    
   

    

     

    
    

     

              self.y_scale=y_scale
     
    
   

    

     

    
    

     

              self.x_scale=x_scale
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
    def forward(self):
     
    
   

    

     

    
    

     

              anchors = []
     
    
   

    

     

    
    

     
        # 'min_sizes': [[16, 32], [64, 128], [256, 512]],
     
    
   

    

     

    
    

     
        # 'steps': [8, 16, 32],
     
    
   

    

     

    
    

     
        #feature_maps:[5,2.5,1.25]
     
    
  
 

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

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