Commit 9054e662 authored by Ross Girshick's avatar Ross Girshick
Browse files

use the new Python layer

parent 8f5c4e7f
name: "CaffeNet" name: "CaffeNet"
input: "data" layer {
input_shape { name: 'data'
dim: 1 type: 'Python'
dim: 3 top: 'data'
dim: 227 top: 'rois'
dim: 227 top: 'labels'
} top: 'bbox_targets'
input: "rois" top: 'bbox_loss_weights'
input_shape { python_param {
dim: 1 # to be changed on-the-fly to num ROIs module: 'roi_data_layer.layer'
dim: 5 # [batch ind, x1, y1, x2, y2] zero-based indexing layer: 'DataLayer'
} param_str: "'num_classes': 21"
input: "labels" }
input_shape {
dim: 1 # to be changed on-the-fly to match num ROIs
}
input: "bbox_targets"
input_shape {
dim: 1 # to be changed on-the-fly to match num ROIs
dim: 84 # 4 * K (=21) classes
}
input: "bbox_loss_weights"
input_shape {
dim: 1 # to be changed on-the-fly to match num ROIs
dim: 84 # 4 * K (=21) classes
} }
layer { layer {
name: "conv1" name: "conv1"
......
name: "VGG_ILSVRC_16_layers" name: "VGG_ILSVRC_16_layers"
layer {
input: "data" name: 'data'
input_shape { type: 'Python'
dim: 1 top: 'data'
dim: 3 top: 'rois'
dim: 224 top: 'labels'
dim: 224 top: 'bbox_targets'
} top: 'bbox_loss_weights'
python_param {
input: "rois" module: 'roi_data_layer.layer'
input_shape { layer: 'DataLayer'
dim: 1 # to be changed on-the-fly to num ROIs param_str: "'num_classes': 21"
dim: 5 # [batch ind, x1, y1, x2, y2] zero-based indexing }
} }
input: "labels"
input_shape {
dim: 1 # to be changed on-the-fly to match num ROIs
}
input: "bbox_targets"
input_shape {
dim: 1 # to be changed on-the-fly to match num ROIs
dim: 84 # 4 * K (=21) classes
}
input: "bbox_loss_weights"
input_shape {
dim: 1 # to be changed on-the-fly to match num ROIs
dim: 84 # 4 * K (=21) classes
}
layer { layer {
name: "conv1_1" name: "conv1_1"
type: "Convolution" type: "Convolution"
......
name: "VGG_CNN_M_1024" name: "VGG_CNN_M_1024"
input: "data" layer {
input_shape { name: 'data'
dim: 1 type: 'Python'
dim: 3 top: 'data'
dim: 224 top: 'rois'
dim: 224 top: 'labels'
} top: 'bbox_targets'
input: "rois" top: 'bbox_loss_weights'
input_shape { python_param {
dim: 1 # to be changed on-the-fly to num ROIs module: 'roi_data_layer.layer'
dim: 5 # [batch ind, x1, y1, x2, y2] zero-based indexing layer: 'DataLayer'
} param_str: "'num_classes': 21"
input: "labels" }
input_shape {
dim: 1 # to be changed on-the-fly to match num ROIs
}
input: "bbox_targets"
input_shape {
dim: 1 # to be changed on-the-fly to match num ROIs
dim: 84 # 4 * K (=21) classes
}
input: "bbox_loss_weights"
input_shape {
dim: 1 # to be changed on-the-fly to match num ROIs
dim: 84 # 4 * K (=21) classes
} }
layer { layer {
name: "conv1" name: "conv1"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment