SSD caffe模型

from __future__ import print_function
import caffe
from caffe.model_libs import *
from google.protobuf import text_format

import math
import os
import shutil
import stat
import subprocess
import sys

# Add extra layers on top of a “base“ network （e.g. VGGNet or Inception）.
def AddExtralayers（net use_batchnorm=True lr_mult=1）:
    use_relu = True

    # Add additional convolutional layers.
    # 19 x 19
    from_layer = net.keys（）[-1]

    # TODO（weiliu89）: Construct the name using the last layer to avoid duplication.
    # 10 x 10
    out_layer = “conv6_1“
    ConvBNlayer（net from_layer out_layer use_batchnorm use_relu 256 1 0 1
        lr_mult=lr_mult）

    from_layer = out_layer
    out_layer = “conv6_2“
    ConvBNlayer（net from_layer out_layer use_batchnorm use_relu 512 3 1 2
        lr_mult=lr_mult）

    # 5 x 5
    from_layer = out_layer
    out_layer = “conv7_1“
    ConvBNlayer（net from_layer out_layer use_batchnorm use_relu 128 1 0 1
      lr_mult=lr_mult）

    from_layer = out_layer
    out_layer = “conv7_2“
    ConvBNlayer（net from_layer out_layer use_batchnorm use_relu 256 3 1 2
      lr_mult=lr_mult）

    # 3 x 3
    from_layer = out_layer
    out_layer = “conv8_1“
    ConvBNlayer（net from_layer out_layer use_batchnorm use_relu 128 1 0 1
      lr_mult=lr_mult）

    from_layer = out_layer
    out_layer = “conv8_2“
    ConvBNlayer（net from_layer out_layer use_batchnorm use_relu 256 3 0 1
      lr_mult=lr_mult）

    # 1 x 1
    from_layer = out_layer
    out_layer = “conv9_1“
    ConvBNlayer（net from_layer out_layer use_batchnorm use_relu 128 1 0 1
      lr_mult=lr_mult）

    from_layer = out_layer
    out_layer = “conv9_2“
    ConvBNlayer（net from_layer out_layer use_batchnorm use_relu 256 3 0 1
      lr_mult=lr_mult）

    return net


### Modify the following parameters accordingly ###
# The directory which contains the caffe code.
# We assume you are running the script at the CAFFE_ROOT.
caffe_root = os.getcwd（）

# Set true if you want to start training right after generating all files.
run_soon = True
# Set true if you want to load from most recently saved snapshot.
# Otherwise we will load from the pretrain_model defined below.
resume_training = True
# If true Remove old model files.
remove_old_models = False

# The database file for training data. Created by data/VOC0712Plus/create_data.sh
train_data = “examples/VOC0712Plus/VOC0712Plus_trainval_lmdb“
# The database file for testing data. Created by data/VOC0712Plus/create_data.sh
test_data = “examples/VOC0712Plus/VOC0712Plus_test_lmdb“
# Specify the batch sampler.
resize_width = 300
resize_height = 300
resize = “{}x{}“.format（resize_width resize_height）
batch_sampler = [
        {
                ‘sampler‘: {
                        }
                ‘max_trials‘: 1
                ‘max_sample‘: 1
        }
        {
                ‘sampler‘: {
                        ‘min_scale‘: 0.3