Skip to content
Snippets Groups Projects
Commit 9054e662 authored by Ross Girshick's avatar Ross Girshick
Browse files

use the new Python layer

parent 8f5c4e7f
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
models/CaffeNet/train.prototxt
+ 13
− 25
View file @ 9054e662
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"
......
models/VGG16/train.prototxt
+ 14
− 32
View file @ 9054e662
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"
......
models/VGG_CNN_M_1024/train.prototxt
+ 13
− 25
View file @ 9054e662
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"
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment