예제 #1
0
 def anchors(self):
     """Get sSD anchors"""
     return ssd_anchors_all_layers(
         self.ssd_params.img_shape, self.ssd_params.feat_shapes,
         self.ssd_params.anchor_sizes, self.ssd_params.anchor_ratios,
         self.ssd_params.anchor_steps, self.ssd_params.anchor_offset,
         np.float32)
예제 #2
0
 def anchors(self, img_shape, dtype=np.float32):
     """
     给定一张输入图,计算所有特征图的默认框
     """
     return ssd_anchors.ssd_anchors_all_layers(
         img_shape, self.params.feat_shapes, self.params.anchor_sizes,
         self.params.anchor_ratios, self.params.anchor_steps,
         self.params.anchor_offset, dtype)
예제 #3
0
 def anchors(self):
     """Get sSD anchors"""
     return ssd_anchors_all_layers(self.ssd_params.img_shape,
                                   self.ssd_params.feat_shapes,
                                   self.ssd_params.anchor_sizes,
                                   self.ssd_params.anchor_ratios,
                                   self.ssd_params.anchor_steps,
                                   self.ssd_params.anchor_offset,
                                   np.float32)