深度学习框架Caffe学习系列(2)：DarkNet 转战 Caffe 经验汇总

最新推荐文章于 2023-08-02 11:28:24 发布

努力努力在努力丶

最新推荐文章于 2023-08-02 11:28:24 发布

阅读量413

点赞数

分类专栏：机器学习与深度学习文章标签： Caffe

本文链接：https://blog.csdn.net/qipeng_master/article/details/85113415

版权

机器学习与深度学习专栏收录该内容

10 篇文章 0 订阅

订阅专栏

--darknet 转战 caffe--

darknet的cfg文件转换成caffe的prototxt文件

> 卷积层重写：

# darknet 的cfg文件
[convolutional]
batch_normalize=1
filters=32
size=3
stride=1
pad=1
activation=leaky

# caffe的prototxt书写
layer {
  name: "conv1"
  type: "Convolution"
  bottom: "data"
  top: "conv1"
  param {
    lr_mult: 1
  }
  param {
    lr_mult: 2
  }
  convolution_param {
    num_output: 32
    pad: 1
    kernel_size: 3
    stride: 1
    weight_filler {
      type: "gaussian"
      std: 0.0001
    }
    bias_filler {
      type: "constant"
    }
  }
}
layer {
    name: "batch_norm1"
    type: "BatchNorm"
    bottom: "conv1"
    top: "conv1"
    batch_norm_param {
        use_global_stats: false
    }
    include {
        phase: TRAIN
    }
}
layer {
    name: "batch_norm1"
    type: "BatchNorm"
    bottom: "conv1"
    top: "conv1"
    batch_norm_param {
        use_global_stats: true
    }
    include {
        phase: TEST
    }
}
layer {
    name: "scale1"
    type: "Scale"
    bottom: "conv1"
    top: "conv1"
    scale_param {
        bias_term: true
    }
}
layer {
  name: "relu1"
  type: "ReLU"
  bottom: "conv1"
  top: "conv1"
  relu_param{
    negative_slope: 0.1
  }
}