Caffe将4个通道图像分类为同一类

时间:2019-06-11 13:56:21

标签: caffe pycaffe

由于无法访问Caffe谷歌论坛,因此我将问题发布在这里。

我有两类要使用caffe分类的图像。这些图像有4个通道,并在训练阶段转换为lmdb。

训练进行得很好,但是,当使用4通道png图像测试模型时,分类器估计总是相同的类。

预先感谢

我试图让变压器读取4个通道的图像,但我不知道我是否正确完成

image = cv2.imread(imgstest_folder,cv2.IMREAD_UNCHANGED).astype(np.float32)/255.0

transformer.set_transpose('data', (2,0,1))
transformer.set_channel_swap('data', (3,2,1,0)) 
transformed_image = transformer.preprocess('data', image)    
net.blobs['data'].data[i] = transformed_image

0 个答案:

没有答案