caffe中通过prototxt将输入通道为RGB转为BGR

最新推荐文章于 2022-07-16 20:24:00 发布

wonengguwozai

最新推荐文章于 2022-07-16 20:24:00 发布

阅读量3.1k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： caffe

本文链接：https://blog.youkuaiyun.com/wonengguwozai/article/details/81298038

caffe 专栏收录该内容

22 篇文章

订阅专栏

本文介绍如何将MXNet训练的模型转换到Caffe平台，并解决因图像读取格式不同（RGB vs BGR）导致的问题。通过在Caffe的prototxt文件中添加特定层，实现从RGB到BGR的通道顺序转换，确保模型正确运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

缘由：在mxnet下训练的模型，通过转换器转到caffe下进行inference，由于mxnet在训练和测试是直接加载的图像的RGB形式，而caffe调用opencv的缘故在读取图像时是通过BGR形式读取的，所以需要进行通道转换。下面方法是直接通过添加prototxt层的形式进行转换。

layer {
name: "convert_slice"
        type: "Slice"
        bottom: "data_"
        top: "B"
        top: "G"
        top: "R"
        slice_param {
          axis: 1
        slice_point: 1
        slice_point: 2
        }
}

layer {
name: "convert_concat"
        type:"Concat"
        bottom:"R"
        bottom:"G"
        bottom:"B"
        top:"data"
}