对于大小为 795 的轴 0，索引 795 超出范围

如何解决对于大小为 795 的轴 0，索引 795 超出范围

我正在尝试解决上述错误，但无法解决。我正在尝试运行以下 python 函数：

定义 plot_test_samples():

fileList = os.listdir('../data/test_samples')
fileList = filter(lambda x: '.png' in x,fileList)
fileList.sort()
case_slices = [ [int(s) for s in fname.replace('.','_').split('_') if s.isdigit()] for fname in fileList]
case_slices = np.array(case_slices)
X_test = np.load('../data/X_test.npy')
n_imgs = np.load('../data/test_n_imgs.npy').cumsum()

img_rows = X_test.shape[1]
img_cols = img_rows
model = get_model(img_rows,img_cols)

n_cols= 4
n_rows = int( np.ceil(len(case_slices)/n_cols*2) )

fig = plt.figure(figsize=[ 4*n_cols,int(4*n_rows)] )
gs = gridspec.GridSpec( n_rows,n_cols )

imgs = [ X_test[n_imgs[row[0]-1]+row[1]] for row in case_slices ]
imgs = np.stack(imgs)
masks = model.predict( imgs,verbose=1)
for mm,row in enumerate(case_slices):
    ax = fig.add_subplot(gs[2*mm])
    img = cv2.imread('../data/test_samples'+'/'+fileList[mm],cv2.IMREAD_COLOR)
    ax.imshow(cv2.cvtColor(img,cv2.COLOR_BGR2RGB) )
    ax = fig.add_subplot(gs[2*mm+1])
    ax.imshow(imgs[mm,:,0],cmap='gray' )
    contours = find_contours(masks[mm,0.01,fully_connected='high')
    for n,contour in enumerate(contours):
        ax.plot(contour[:,1],contour[:,linewidth=1,color='b')

fig.savefig('../images/test_samples.png',bBox_inches='tight',dpi=300 )



len case_slices=795
    shape(case_slices)=(795,1)
    len n_imgs=30
    len X_test=795
    shape(X_test)=(795,256,1)

请任何人知道我如何解决这个问题让我知道，提前谢谢

解决方法

该错误基本上是在告诉您，您正在尝试从具有 795 个数字的数组中访问编号为 795 的元素，该数组从位置编号 0 到位置编号 794。

尝试将您的代码更改为：

len case_slices=795
shape(case_slices)=(794,1)
len n_imgs=30
len X_test=795
shape(X_test)=(794,256,1)