将 caffe 上的净转发从 MATLAB 转换为 Python 时出现问题

Question

我正在尝试将一些 MATLAB 代码转换为 Python，因为我在 MATLAB 上运行编译代码不成功（每当我运行它时它就会崩溃）。到目前为止，我已经完成了代码转换和运行但结果是非常错误的。所以我尝试调试它，但部分代码仍然给我带来不确定性，因为我无法在 MATLAB

上检查结果

这是我要转换的 MATLAB 代码

%Prepare images
im=reshape(im,[size(im)]); im=single(im)/255;
im_data = im(:, :, [3, 2, 1]);  % permute channels from RGB to BGR
im_data = permute(im_data, [2, 1, 3]); 

%pass images  
out_im = net.forward({im_data});
n_out=out_im{2}; al_out=out_im{1}; light_out=out_im{3};

这是我创建的 Python 代码

#Prepare images (rotate,flip,change color,reshape)
im=np.reshape(im,(im.shape))
im=np.float32(im)/255
#already convert to BGR on top
#im_data = cv2.cvtColor(im, cv2.COLOR_RGB2BGR)
#Dibalik lalu di rotate 90 ke kiri
im_data = np.transpose(im, (1, 0, 2)) 
im_data=np.moveaxis(im_data, -1, 0) 
im_input = im_data[np.newaxis, : , :, :]
net.blobs['data'].reshape(*im_input.shape)
net.blobs['data'].data[...] = im_input
#pass images  
out_im = net.forward()
n_out=out_im['Nconv0'][0] 
al_out=out_im['Aconv0'][0] 
light_out=out_im['fc_light'][0]

我尝试按照 caffe 用户的教程进行操作，输入应该是 (N,channel,width,height) 所以我将输入转换为 (1,3,128,128) 但是在 MATLAB 代码中我认为形状是（宽度、高度、通道）。上次我尝试仅输入 3 维时出现错误。

Answer 1

转置时是(2,0,1)，不是(1,0,2)。

inputs = cv.resize(inputs,(128,128))
inputs = inputs.transpose((2,0,1))
inputs = inputs[None,:]
net.forward_all( data = inputs )

试试这个

Answer 2

inputs = cv.resize(inputs,(128,128))
inputs = inputs * 0.00390625
inputs = inputs.transpose((2,0,1))
inputs = inputs[None,:]
net.forward_all( data = inputs )

再试一次

将 caffe 上的净转发从 MATLAB 转换为 Python 时出现问题

Problem converting net forward on caffe from MATLAB to Python

python

matlab

caffe