anchors_plane python

  

   

    

     

    
    

     

      def anchors_plane(feat_h, feat_w, stride, base_anchors):
     
    
   

    

     

    
    

     
 # return anchors_cython(feat_h, feat_w, stride, base_anchor)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       A = base_anchors.shape[0]
     
    
   

    

     

    
    

     

       all_anchors = np.zeros((feat_h, feat_w, A, 4))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 for iw in range(feat_w):
     
    
   

    

     

    
    

     

       sw = iw * stride
     
    
   

    

     

    
    

     
 for ih in range(feat_h):
     
    
   

    

     

    
    

     

       sh = ih * stride
     
    
   

    

     

    
    

     
 for k in range(A):
     
    
   

    

     

    
    

     

       all_anchors[ih, iw, k, 0] = base_anchors[k, 0] + sw
     
    
   

    

     

    
    

     

       all_anchors[ih, iw, k, 1] = base_anchors[k, 1] + sh
     
    
   

    

     

    
    

     

       all_anchors[ih, iw, k, 2] = base_anchors[k, 2] + sw
     
    
   

    

     

    
    

     

       all_anchors[ih, iw, k, 3] = base_anchors[k, 3] + sh
     
    
   

    

     

    
    

     
 return all_anchors
     
    
  
 

cpython:

  

   

    

     

    
    

     

      cimport cython
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      cimport numpy as np
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      DTYPE = np.float32
     
    
   

    

     

    
    

     

      ctypedef np.float32_t DTYPE_t
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def anchors_cython(int height, int width, int stride, np.ndarray[DTYPE_t, ndim=2] ba
     
    
  
 

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

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