- 缘由:在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" }
caffe中通过prototxt将输入通道为RGB转为BGR
最新推荐文章于 2022-07-16 20:24:00 发布