c++中opencv调用caffe ,输入blob设置问题。

一般的opencv 调用caffe feed 数据都是用blobFromImage()但是这只能解决输入是1*3*244*244单张图片,如果要是有特殊需求,比如想要输入1*6*244*244该怎么解决,这时候就不能用上面API了,只能自己构建一个大小为(1*6*244*244)的blob,还要一点需要注意的是,在原始里面caffe输入要求是float数据,千万不要认为什么事,就私自把数据改成int的丢进去,这样出来的结果可能会非常奇怪。

构建方式如下:

Mat M;

vector<int> input_size = { 1,6,224,224 };   // create dims 1*6*224*224

M.create(input_size, CV_32FC1);

M赋值操作:

void  meage(Mat &img1, Mat &img2,Mat &img)//1*6*224*224,其中img1,img2都是大小为224*224*3的图片,img==M
{

    for (int i = 0; i < img.size[2]; i++)//224
        for (int j = 0; j < img.size[3]; j++)//4
        {
            img.at<Vec4f>(0,0,i)[j]= img1.at<Vec3b>(i, j)[0]-128; //千万注意<Vec4f>这里,楼主的输入要求是float,所以这里一定是float,不能是<Vec4i>或者<Vec4b>这种
            img.at<Vec4f>(0,1,i)[j] = img1.at<Vec3b>(i, j)[1]-128;//通道1,位置(i,j)
            img.at<Vec4f>(0,2,i)[j] = img1.at<Vec3b>(i, j)[2]-128;
            img.at<Vec4f>(0,3,i)[j] = img2.at<Vec3b>(i, j)[0]-128;
            img.at<Vec4f>(0,4, i)[j] = img2.at<Vec3b>(i, j)[1]-128;
            img.at<Vec4f>(0,5,i)[j] = img2.at<Vec3b>(i, j)[2]-128;
        }

}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值