Skip to content

Instantly share code, notes, and snippets.

input: "data"
input_shape {
dim: 1
dim: 3
dim: 112
dim: 112
}
layer {
name: "conv1"
type: "Convolution"
name: "ResNet-50"
layer {
name: 'input-data'
type: 'Python'
top: 'data'
top: 'im_info'
top: 'gt_boxes'
python_param {
module: 'roi_data_layer.layer'
layer: 'RoIDataLayer'
name: "GoogleNet"
layer {
name: "data"
type: "Data"
top: "data"
top: "label"
include {
phase: TRAIN
}
transform_param {
layer {
name: "data"
type: "DriveData"
top: "data"
top: "label"
top: "label_type"
include:{ phase: TRAIN }
transform_param {
mean_file: "./../../data/lmdb/train_mean.binaryproto"
}
name: "Inception_Resnet2_Imagenet"
layer {
name: "data"
type: "Data"
top: "data"
top: "Orientation_label"
top: "Color_label"
top: "Model_label"
include {
phase: TRAIN
name: "gan_newface"
layer {
name: "images"
type: "ImageData"
top: "face_images"
top: "label"
include {
phase: TRAIN
}
transform_param {
name: "GAN"
layer {
name: 'input_data'
type: 'Python'
top: 'gray_image'
top: 'color_image'
top: 'label'
top: 'label_g'
python_param {
module: 'data_layer'
name: "ResNet"
layer {
name: "data"
type: "Input"
top: "data"
input_param { shape: { dim: 1 dim: 3 dim: 224 dim: 224 } }
}
layer {
name: "bn_data"
name: "AlexNet"
layer {
name: "data"
type: "Data"
top: "data"
top: "label"
include {
phase: TRAIN
}
transform_param {
name:nichaoNet
layer {
name: "data"
type: "DriveData"
top: "data"
top: "label"
top: "label_type"
include:{ phase: TRAIN }
transform_param {
mean_file: "./../../data/lmdb/train_mean.binaryproto"