VGG_19 train_vali.prototxt file

简介: name: "VGG_ILSVRC_19_layer"layer {  name: "data"  type: "ImageData"  top: "data"  top: "label"  include {    phase: TRAIN  }   image_data_param {    batch_size: 12    source: ".
name: "VGG_ILSVRC_19_layer"

layer {
  name: "data"
  type: "ImageData"
  top: "data"
  top: "label"
  include {
    phase: TRAIN
  }
 
  image_data_param {
    batch_size: 12
    source: "../../fine_tuning_data/HAT_fineTuning_data/train_data_fineTuning.txt"
    root_folder: "../../fine_tuning_data/HAT_fineTuning_data/train_data/"
  }
}

layer {
  name: "data"
  type: "ImageData"
  top: "data"
  top: "label"
  include {
    phase: TEST
  }
  transform_param {
    mirror: false
  }
  image_data_param {
    batch_size: 10
    source: "../../fine_tuning_data/HAT_fineTuning_data/test_data_fineTuning.txt"
    root_folder: "../../fine_tuning_data/HAT_fineTuning_data/test_data/"
  }
}

layer {
  bottom:"data"
  top:"conv1_1"
  name:"conv1_1"
  type:"Convolution"
  convolution_param {
    num_output:64
    pad:1
    kernel_size:3
  }
}
layer {
  bottom:"conv1_1"
  top:"conv1_1"
  name:"relu1_1"
  type:"ReLU"
}
layer {
  bottom:"conv1_1"
  top:"conv1_2"
  name:"conv1_2"
  type:"Convolution"
  convolution_param {
    num_output:64
    pad:1
    kernel_size:3
  }
}
layer {
  bottom:"conv1_2"
  top:"conv1_2"
  name:"relu1_2"
  type:"ReLU"
}
layer {
  bottom:"conv1_2"
  top:"pool1"
  name:"pool1"
  type:"Pooling"
  pooling_param {
    pool:MAX
    kernel_size:2
    stride:2
  }
}
layer {
  bottom:"pool1"
  top:"conv2_1"
  name:"conv2_1"
  type:"Convolution"
  convolution_param {
    num_output:128
    pad:1
    kernel_size:3
  }
}
layer {
  bottom:"conv2_1"
  top:"conv2_1"
  name:"relu2_1"
  type:"ReLU"
}
layer {
  bottom:"conv2_1"
  top:"conv2_2"
  name:"conv2_2"
  type:"Convolution"
  convolution_param {
    num_output:128
    pad:1
    kernel_size:3
  }
}
layer {
  bottom:"conv2_2"
  top:"conv2_2"
  name:"relu2_2"
  type:"ReLU"
}
layer {
  bottom:"conv2_2"
  top:"pool2"
  name:"pool2"
  type:"Pooling"
  pooling_param {
    pool:MAX
    kernel_size:2
    stride:2
  }
}
layer {
  bottom:"pool2"
  top:"conv3_1"
  name: "conv3_1"
  type:"Convolution"
  convolution_param {
    num_output:256
    pad:1
    kernel_size:3
  }
}
layer {
  bottom:"conv3_1"
  top:"conv3_1"
  name:"relu3_1"
  type:"ReLU"
}
layer {
  bottom:"conv3_1"
  top:"conv3_2"
  name:"conv3_2"
  type:"Convolution"
  convolution_param {
    num_output:256
    pad:1
    kernel_size:3
  }
}
layer {
  bottom:"conv3_2"
  top:"conv3_2"
  name:"relu3_2"
  type:"ReLU"
}
layer {
  bottom:"conv3_2"
  top:"conv3_3"
  name:"conv3_3"
  type:"Convolution"
  convolution_param {
    num_output:256
    pad:1
    kernel_size:3
  }
}
layer {
  bottom:"conv3_3"
  top:"conv3_3"
  name:"relu3_3"
  type:"ReLU"
}
layer {
  bottom:"conv3_3"
  top:"conv3_4"
  name:"conv3_4"
  type:"Convolution"
  convolution_param {
    num_output:256
    pad:1
    kernel_size:3
  }
}
layer {
  bottom:"conv3_4"
  top:"conv3_4"
  name:"relu3_4"
  type:"ReLU"
}
layer {
  bottom:"conv3_4"
  top:"pool3"
  name:"pool3"
  type:"Pooling"
  pooling_param {
    pool:MAX
    kernel_size: 2
    stride: 2
  }
}
layer {
  bottom:"pool3"
  top:"conv4_1"
  name:"conv4_1"
  type:"Convolution"
  convolution_param {
    num_output: 512
    pad: 1
    kernel_size: 3
  }
}
layer {
  bottom:"conv4_1"
  top:"conv4_1"
  name:"relu4_1"
  type:"ReLU"
}
layer {
  bottom:"conv4_1"
  top:"conv4_2"
  name:"conv4_2"
  type:"Convolution"
  convolution_param {
    num_output: 512
    pad: 1
    kernel_size: 3
  }
}
layer {
  bottom:"conv4_2"
  top:"conv4_2"
  name:"relu4_2"
  type:"ReLU"
}
layer {
  bottom:"conv4_2"
  top:"conv4_3"
  name:"conv4_3"
  type:"Convolution"
  convolution_param {
    num_output: 512
    pad: 1
    kernel_size: 3
  }
}
layer {
  bottom:"conv4_3"
  top:"conv4_3"
  name:"relu4_3"
  type:"ReLU"
}
layer {
  bottom:"conv4_3"
  top:"conv4_4"
  name:"conv4_4"
  type:"Convolution"
  convolution_param {
    num_output: 512
    pad: 1
    kernel_size: 3
  }
}
layer {
  bottom:"conv4_4"
  top:"conv4_4"
  name:"relu4_4"
  type:"ReLU"
}
layer {
  bottom:"conv4_4"
  top:"pool4"
  name:"pool4"
  type:"Pooling"
  pooling_param {
    pool:MAX
    kernel_size: 2
    stride: 2
  }
}
layer {
  bottom:"pool4"
  top:"conv5_1"
  name:"conv5_1"
  type:"Convolution"
  convolution_param {
    num_output: 512
    pad: 1
    kernel_size: 3
  }
}
layer {
  bottom:"conv5_1"
  top:"conv5_1"
  name:"relu5_1"
  type:"ReLU"
}
layer {
  bottom:"conv5_1"
  top:"conv5_2"
  name:"conv5_2"
  type:"Convolution"
  convolution_param {
    num_output: 512
    pad: 1
    kernel_size: 3
  }
}
layer {
  bottom:"conv5_2"
  top:"conv5_2"
  name:"relu5_2"
  type:"ReLU"
}
layer {
  bottom:"conv5_2"
  top:"conv5_3"
  name:"conv5_3"
  type:"Convolution"
  convolution_param {
    num_output: 512
    pad: 1
    kernel_size: 3
  }
}
layer {
  bottom:"conv5_3"
  top:"conv5_3"
  name:"relu5_3"
  type:"ReLU"
}
layer {
  bottom:"conv5_3"
  top:"conv5_4"
  name:"conv5_4"
  type:"Convolution"
  convolution_param {
    num_output: 512
    pad: 1
    kernel_size: 3
  }
}
layer {
  bottom:"conv5_4"
  top:"conv5_4"
  name:"relu5_4"
  type:"ReLU"
}
layer {
  bottom:"conv5_4"
  top:"pool5"
  name:"pool5"
  type:"Pooling"
  pooling_param {
    pool:MAX
    kernel_size: 2
    stride: 2
  }
}
layer {
  bottom:"pool5"
  top:"fc6_"
  name:"fc6_"
  type:"InnerProduct"
  inner_product_param {
    num_output: 4096
  }
}
layer {
  bottom:"fc6_"
  top:"fc6_"
  name:"relu6"
  type:"ReLU"
}
layer {
  bottom:"fc6_"
  top:"fc6_"
  name:"drop6"
  type:"Dropout"
  dropout_param {
    dropout_ratio: 0.5
  }
}
layer {
  bottom:"fc6_"
  top:"fc7"
  name:"fc7"
  type:"InnerProduct"
  inner_product_param {
    num_output: 4096
  }
}
layer {
  bottom:"fc7"
  top:"fc7"
  name:"relu7"
  type:"ReLU"
}
layer {
  bottom:"fc7"
  top:"fc7"
  name:"drop7"
  type:"Dropout"
  dropout_param {
    dropout_ratio: 0.5
  }
}
layer {
  bottom:"fc7"
  top:"fc8_"
  name:"fc8_"
  type:"InnerProduct"
  inner_product_param {
    num_output: 27
  }
}

layer {
  name: "sigmoid"
  type: "Sigmoid"
  bottom: "fc8_"
  top: "fc8_"
}

 layer {
   name: "accuracy"
   type: "Accuracy"
   bottom: "fc8_"
   bottom: "label"
   top: "accuracy"
   include {
     phase: TEST
   }
 }

layer {
  name: "loss"
  type: "EuclideanLoss"
  bottom: "fc8_"
  bottom: "label"
  top: "loss"
}



 

相关文章
|
1月前
|
机器学习/深度学习 资源调度 监控
PyTorch使用Tricks:Dropout,R-Dropout和Multi-Sample Dropout等 !!
PyTorch使用Tricks:Dropout,R-Dropout和Multi-Sample Dropout等 !!
25 0
|
4月前
|
机器学习/深度学习 算法 数据可视化
模型训练(Model Training)
模型训练(Model Training)是指使用数据集对模型进行训练,使其能够从数据中学习到特征和模式,进而完成特定的任务。在深度学习领域,通常使用反向传播算法来训练模型,其中模型会根据数据集中的输入和输出,不断更新其参数,以最小化损失函数。
105 1
|
4月前
|
机器学习/深度学习 自然语言处理 PyTorch
Model Inference
模型推理(Model Inference)是指使用已经训练好的机器学习模型来对新数据进行预测或分类的过程。模型推理是机器学习中的一个重要环节,其目的是利用训练好的模型对新数据进行预测或分类,从而得到结果。
71 1
|
8月前
|
机器学习/深度学习 并行计算 PyTorch
【PyTorch】Training Model
【PyTorch】Training Model
64 0
|
机器学习/深度学习 PyTorch 算法框架/工具
Pytorch和DCGAN生成肖像画(下)
Pytorch和DCGAN生成肖像画
123 0
Pytorch和DCGAN生成肖像画(下)
|
机器学习/深度学习 PyTorch 算法框架/工具
Pytorch和DCGAN生成肖像画(上)
Pytorch和DCGAN生成肖像画
114 0
Pytorch和DCGAN生成肖像画(上)
|
编解码 PyTorch 算法框架/工具
YOLOv5的Tricks | 【Trick3】Test Time Augmentation(TTA)
一句话简单的介绍Test Time Augmentation(TTA)就是测试过程中也使用数据增强,官方教程介绍:Test-Time Augmentation (TTA) Tutorial
410 0
|
机器学习/深度学习
神经网络与深度学习---train_loss和val_loss(test_lost)分析
神经网络与深度学习---train_loss和val_loss(test_lost)分析
2211 2
|
TensorFlow 算法框架/工具
TF之DCGAN:基于TF利用DCGAN测试MNIST数据集并进行生成过程全记录
TF之DCGAN:基于TF利用DCGAN测试MNIST数据集并进行生成过程全记录
TF之DCGAN:基于TF利用DCGAN测试MNIST数据集并进行生成过程全记录
DL之DCGNN:基于TF利用DCGAN实现在MNIST数据集上训练生成新样本
DL之DCGNN:基于TF利用DCGAN实现在MNIST数据集上训练生成新样本

热门文章

最新文章