name: "tiny_yolo" layer { name: "data" type: "BoxData" top: "data" top: "label" include { phase: TRAIN } data_param { source: "../../data/yolo/lmdb/trainval_lmdb" batch_size: 64 side: 5 backend: LMDB } } layer { name: "data" type: "BoxData" top: "data" top: "label" include { phase: TEST } data_param { source: "../../data/yolo/lmdb/test_lmdb" batch_size: 1 side: 5 backend: LMDB } } layer { name: "conv1" type: "Convolution" bottom: "data" top: "conv1" param { lr_mult: 1 decay_mult: 1 } convolution_param { num_output: 16 pad: 1 kernel_size: 3 stride: 1 bias_term: false weight_filler { type: "xavier" } } } layer { name: "bn1" type: "BatchNorm" bottom: "conv1" top: "bn1" include { phase: TRAIN } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } batch_norm_param { use_global_stats: false } } layer { name: "bn1" type: "BatchNorm" bottom: "conv1" top: "bn1" include { phase: TEST } batch_norm_param { use_global_stats: true } } layer { name: "scale1" type: "Scale" bottom: "bn1" top: "scale1" param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } scale_param { bias_term: true } } layer { name: "relu1" type: "ReLU" bottom: "scale1" top: "scale1" scale_param { bias_term: true } } layer { name: "pool1" type: "Pooling" bottom: "scale1" top: "pool1" pooling_param { pool: MAX kernel_size: 2 stride: 2 } } layer { name: "conv2" type: "Convolution" bottom: "pool1" top: "conv2" param { lr_mult: 1 decay_mult: 1 } convolution_param { num_output: 32 kernel_size: 3 stride: 1 pad: 1 bias_term: false weight_filler { type: "xavier" } } } layer { name: "bn2" type: "BatchNorm" bottom: "conv2" top: "bn2" include { phase: TRAIN } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } batch_norm_param { use_global_stats: false } } layer { name: "bn2" type: "BatchNorm" bottom: "conv2" top: "bn2" include { phase: TEST } batch_norm_param { use_global_stats: true } } layer { name: "scale2" type: "Scale" bottom: "bn2" top: "scale2" param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } scale_param { bias_term: true } } layer { name: "relu2" type: "ReLU" bottom: "scale2" top: "scale2" relu_param { negative_slope: 0.1 } } layer { name: "pool2" type: "Pooling" bottom: "scale2" top: "pool2" pooling_param { pool: MAX kernel_size: 2 stride: 2 } } layer { name: "conv3" type: "Convolution" bottom: "pool2" top: "conv3" param { lr_mult: 1 decay_mult: 1 } convolution_param { num_output: 64 kernel_size: 3 stride: 1 pad: 1 bias_term: false weight_filler { type: "xavier" } } } layer { name: "bn3" type: "BatchNorm" bottom: "conv3" top: "bn3" include { phase: TRAIN } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } batch_norm_param { use_global_stats: false } } layer { name: "bn3" type: "BatchNorm" bottom: "conv3" top: "bn3" include { phase: TEST } batch_norm_param { use_global_stats: true } } layer { name: "scale3" type: "Scale" bottom: "bn3" top: "scale3" param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } scale_param { bias_term: true } } layer { name: "relu3" type: "ReLU" bottom: "scale3" top: "scale3" relu_param { negative_slope: 0.1 } } layer { name: "pool3" type: "Pooling" bottom: "scale3" top: "pool3" pooling_param { pool: MAX kernel_size: 2 stride: 2 } } layer { name: "conv4" type: "Convolution" bottom: "pool3" top: "conv4" param { lr_mult: 1 decay_mult: 1 } convolution_param { num_output: 128 kernel_size: 3 stride: 1 pad: 1 bias_term: false weight_filler { type: "xavier" } } } layer { name: "bn4" type: "BatchNorm" bottom: "conv4" top: "bn4" include { phase: TRAIN } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } batch_norm_param { use_global_stats: false } } layer { name: "bn4" type: "BatchNorm" bottom: "conv4" top: "bn4" include { phase: TEST } batch_norm_param { use_global_stats: true } } layer { name: "scale4" type: "Scale" bottom: "bn4" top: "scale4" param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } scale_param { bias_term: true } } layer { name: "relu4" type: "ReLU" bottom: "scale4" top: "scale4" relu_param { negative_slope: 0.1 } } layer { name: "pool4" type: "Pooling" bottom: "scale4" top: "pool4" pooling_param { pool: MAX kernel_size: 2 stride: 2 } } layer { name: "conv5" type: "Convolution" bottom: "pool4" top: "conv5" param { lr_mult: 1 decay_mult: 1 } convolution_param { num_output: 256 kernel_size: 3 stride: 1 pad: 1 bias_term: false weight_filler { type: "xavier" } } } layer { name: "bn5" type: "BatchNorm" bottom: "conv5" top: "bn5" include { phase: TRAIN } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } batch_norm_param { use_global_stats: false } } layer { name: "bn5" type: "BatchNorm" bottom: "conv5" top: "bn5" include { phase: TEST } batch_norm_param { use_global_stats: true } } layer { name: "scale5" type: "Scale" bottom: "bn5" top: "scale5" param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } scale_param { bias_term: true } } layer { name: "relu5" type: "ReLU" bottom: "scale5" top: "scale5" relu_param { negative_slope: 0.1 } } layer { name: "pool5" type: "Pooling" bottom: "scale5" top: "pool5" pooling_param { pool: MAX kernel_size: 2 stride: 2 } } layer { name: "conv6" type: "Convolution" bottom: "pool5" top: "conv6" param { lr_mult: 1 decay_mult: 1 } convolution_param { num_output: 512 kernel_size: 3 stride: 1 pad: 1 bias_term: false weight_filler { type: "xavier" } } } layer { name: "bn6" type: "BatchNorm" bottom: "conv6" top: "bn6" include { phase: TRAIN } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } batch_norm_param { use_global_stats: false } } layer { name: "bn6" type: "BatchNorm" bottom: "conv6" top: "bn6" include { phase: TEST } batch_norm_param { use_global_stats: true } } layer { name: "scale6" type: "Scale" bottom: "bn6" top: "scale6" param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } scale_param { bias_term: true } } layer { name: "relu6" type: "ReLU" bottom: "scale6" top: "scale6" relu_param { negative_slope: 0.1 } } layer { name: "pool6" type: "Pooling" bottom: "scale6" top: "pool6" pooling_param { pool: MAX kernel_size: 2 stride: 2 } } layer { name: "conv7" type: "Convolution" bottom: "pool6" top: "conv7" param { lr_mult: 1 decay_mult: 1 } convolution_param { num_output: 1024 kernel_size: 3 pad: 1 bias_term: false weight_filler { type: "xavier" } } } layer { name: "bn7" type: "BatchNorm" bottom: "conv7" top: "bn7" include { phase: TRAIN } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } batch_norm_param { use_global_stats: false } } layer { name: "bn7" type: "BatchNorm" bottom: "conv7" top: "bn7" include { phase: TEST } batch_norm_param { use_global_stats: true } } layer { name: "scale7" type: "Scale" bottom: "bn7" top: "scale7" param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } scale_param { bias_term: true } } layer { name: "relu7" type: "ReLU" bottom: "scale7" top: "scale7" relu_param { negative_slope: 0.1 } } layer { name: "conv8" type: "Convolution" bottom: "scale7" top: "conv8" param { lr_mult: 1 decay_mult: 1 } convolution_param { num_output: 256 kernel_size: 3 pad: 1 bias_term: false weight_filler { type: "xavier" } } } layer { name: "bn8" type: "BatchNorm" bottom: "conv8" top: "bn8" include { phase: TRAIN } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } batch_norm_param { use_global_stats: false } } layer { name: "bn8" type: "BatchNorm" bottom: "conv8" top: "bn8" include { phase: TEST } batch_norm_param { use_global_stats: true } } layer { name: "scale8" type: "Scale" bottom: "bn8" top: "scale8" param { lr_mult: 0 decay_mult: 0 } param { lr_mult: 0 decay_mult: 0 } scale_param { bias_term: true } } layer { name: "relu8" type: "ReLU" bottom: "scale8" top: "scale8" relu_param { negative_slope: 0.1 } } layer { name: "fc9" type: "InnerProduct" bottom: "scale8" top: "result" param { lr_mult: 1 decay_mult: 1 } param { lr_mult: 2 decay_mult: 0 } inner_product_param { num_output: 275 weight_filler { type: "gaussian" std: 0.01 } bias_filler { type: "constant" value: 0 } } } layer { name: "det_loss" type: "DetectionLoss" bottom: "result" bottom: "label" top: "det_loss" loss_weight: 1 detection_loss_param { side: 5 num_class: 1 num_object: 2 object_scale: 5.0 noobject_scale: 1.0 class_scale: 1.0 coord_scale: 1.0 sqrt: true constriant: true } }