1. 程式人生 > >caffe學習筆記:caffe網路模型檔案(.prototxt)layer與layers區別

caffe學習筆記:caffe網路模型檔案(.prototxt)layer與layers區別

編寫caffe網路模型檔案的時,有layer與layers兩種方式,其不同之處主要在於網路型別type的定義方式不同,例如定義type為hdf5資料型別時,其定義方式如下。

1. 使用layer

layer {
  name: "data"
  type: "HDF5Data"
  top: "data"
  top: "label"
  hdf5_data_param {
    source: "examples/hdf5_classification/data/train.txt"
    batch_size: 10
  }
}

2. 使用layers

layers {
  name: "data"
  type: HDF5_DATA
  top: "data"
  top: "label"
  hdf5_data_param {
    source: "examples/hdf5_classification/data/train.txt"
    batch_size: 10
  }
}

當需要將layer更改為layers時,其type對應關係如下:

layer:"HDF5Data"、"Convolution"、"ReLU"、"Pooling"、"LRN"、"Dropout"、"InnerProduct"、"EuclideanLoss"

layers:HDF5_DATA、CONVOLUTION、RELU、POOLING、LRN、DROPOUT、INNER_PRODUCT、EUCLIDEAN_LOSS、