keras2.x 实现ssd300

1. from keras.layers import Convolution2D 替换为from keras.layers import Conv2D

# conv2_2(None,150,150,128)
net['conv2_2'] = Conv2D(128, kernel_size=3,
                               activation='relu',
                               padding='same',
                               name='conv2_2')(net['conv2_1'])

2. from keras.layers import AtrousConvolution2D 替换为

net['fc6'] = Conv2D(1024,kernel_size=3, dilation_rate=(6, 6),
                                     activation='relu', padding='same',
                                     name='fc6')(net['pool5'])

3. from keras.layers import merge 替换为from keras.layers import concatenate

net['mbox_priorbox'] = concatenate([net['conv4_3_norm_mbox_priorbox'],
                                  net['fc7_mbox_priorbox'],
                                  net['conv6_2_mbox_priorbox'],
                                  net['conv7_2_mbox_priorbox'],
                                  net['conv8_2_mbox_priorbox'],
                                  net['pool6_mbox_priorbox']],
                                  axis=1,name='mbox_priorbox')

4. PriorBox 重构函数

# def get_output_shape_for(self, input_shape):
    def compute_output_shape(self, input_shape):
        num_priors_ = len(self.aspect_ratios)
        layer_width = input_shape[self.waxis]
        layer_height = input_shape[self.haxis]
        num_boxes = num_priors_ * layer_width * layer_height
        return (input_shape[0], num_boxes, 8)