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

use the new Python layer

parent 8f5c4e7f
name: "CaffeNet"
input: "data"
input_shape {
dim: 1
dim: 3
dim: 227
dim: 227
}
input: "rois"
input_shape {
dim: 1 # to be changed on-the-fly to num ROIs
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 {
name: 'data'
type: 'Python'
top: 'data'
top: 'rois'
top: 'labels'
top: 'bbox_targets'
top: 'bbox_loss_weights'
python_param {
module: 'roi_data_layer.layer'
layer: 'DataLayer'
param_str: "'num_classes': 21"
}
}
layer {
name: "conv1"
......
name: "VGG_ILSVRC_16_layers"
input: "data"
input_shape {
dim: 1
dim: 3
dim: 224
dim: 224
}
input: "rois"
input_shape {
dim: 1 # to be changed on-the-fly to num ROIs
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 {
name: 'data'
type: 'Python'
top: 'data'
top: 'rois'
top: 'labels'
top: 'bbox_targets'
top: 'bbox_loss_weights'
python_param {
module: 'roi_data_layer.layer'
layer: 'DataLayer'
param_str: "'num_classes': 21"
}
}
layer {
name: "conv1_1"
type: "Convolution"
......
name: "VGG_CNN_M_1024"
input: "data"
input_shape {
dim: 1
dim: 3
dim: 224
dim: 224
}
input: "rois"
input_shape {
dim: 1 # to be changed on-the-fly to num ROIs
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 {
name: 'data'
type: 'Python'
top: 'data'
top: 'rois'
top: 'labels'
top: 'bbox_targets'
top: 'bbox_loss_weights'
python_param {
module: 'roi_data_layer.layer'
layer: 'DataLayer'
param_str: "'num_classes': 21"
}
}
layer {
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