2 little 0ne little_0ne 于 2017.09.07 16:26 提问

np的二维数组想要写成lmdb喂给caffe数据层时np的维度应该怎么调

各位大佬是这样的
我现在有一个np数组,是13200x2500的,也就是有13200个训练数据,现在要放到caffe里跑,不过首先要写成lmdb然后喂给数据层。我想把每个数据都当成50x50的图片来处理。
现在有一个问题,我的np是要reshape成 13200 x 1 x 50 x 50的数组然后写回去,在caffe的数据层里设置num_channels: 1,这样吗?_ 还是reshape成13200 x 50 x 50写成lmdb就好,这个具体格式不是很明白

Csdn user default icon
上传中...
上传图片
插入图片