PSMNet/finetune.py出错

File "/home/ubuntu/wgp/PSMNet/finetune.py", line 139, in test   true_disp[index[0][:], index[1][:], index[2][:]] - pred_disp[index[0][:], index[1][:], index[2][:]]) IndexError: index 192 is out of bounds for dimension 1 with size 1

解决办法

    disp_true[index[0][:], index[1][:], index[2][:]] = np.abs(
        true_disp[index[0][:], index[1][:], index[2][:]] - pred_disp[index[0][:], index[1][:], index[2][:]])
    改为
        disp_true[index[0][:], index[1][:], index[2][:]] = np.abs(
        true_disp[index[0][:], index[1][:], index[2][:]] - pred_disp[0,index[0][:], index[1][:], index[2][:]])