我为什么会收到这个IndexError?

时间:2020-08-11 19:54:59

标签: python matplotlib

我已将我的签名文件夹拆分为训练,验证和测试。现在,我想看到随机选择了两个orig_train签名和一个forg_val签名。但我得到错误。 注意:我没有在训练中使用伪造的签名(这是我研究的一部分,不是我错误地做的事情)

#Train-Validation-Test Split
#Signatures of 4 people are used for training
#Signatures of 2 people are used for validation
#Signatures of 2 people are used for testing

orig_train, orig_val, orig_test = orig_groups[:4], orig_groups[4:6], orig_groups[6:]
forg_val, forg_test = forg_groups[:4], forg_groups[4:]```
def visualize_sample_signature():
    '''Function to randomly select a signature from train set and
    print two genuine copies and one forged copy'''
    fig, (ax1, ax2, ax3) = plt.subplots(1, 3, figsize = (10, 10))
    k = np.random.randint(len(orig_train))
    orig_img_names = random.sample(orig_train[k], 1)
    forg_img_name = random.sample(forg_val[k], 0)
    orig_img1 = cv2.imread(orig_img_names[0], 0)
    orig_img2 = cv2.imread(orig_img_names[1], 0)
    forg_img = plt.imread(forg_img_name[0], 0)
    orig_img1 = cv2.resize(orig_img1, (img_w, img_h))
    orig_img2 = cv2.resize(orig_img2, (img_w, img_h))
    forg_img = cv2.resize(forg_img, (img_w, img_h))

    ax1.imshow(orig_img1, cmap = 'gray')
    ax2.imshow(orig_img2, cmap = 'gray')
    ax3.imshow(forg_img, cmap = 'gray')

    ax1.set_title('Genuine Copy')
    ax1.axis('off')
    ax2.set_title('Genuine Copy')
    ax2.axis('off')
    ax3.set_title('Forged Copy')
    ax3.axis('off')

visualize_sample_signature()
IndexError                                Traceback (most recent call last)
<ipython-input-113-8ca6484a3a89> in <module>
----> 1 visualize_sample_signature()

<ipython-input-112-2588ee191bd7> in visualize_sample_signature()
      7     forg_img_name = random.sample(forg_val[k], 0)
      8     orig_img1 = cv2.imread(orig_img_names[0], 0)
----> 9     orig_img2 = cv2.imread(orig_img_names[1], 0)
     10     forg_img = plt.imread(forg_img_name[0], 0)
     11     orig_img1 = cv2.resize(orig_img1, (img_w, img_h))

IndexError: list index out of range

1 个答案:

答案 0 :(得分:0)

调用random.sample()时

orig_img_names = random.sample(orig_train[k], 1)

您仅绘制一个样本。

语法:random.sample(sequence,k)

顺序:这是您要选择的内容

k:是您选择的样本数

在这种情况下,您选择1个样本,然后在代码中稍后要求orig_img_names中的下一个元素。

orig_img2 = cv2.imread(orig_img_names[1], 0)

位置1没有元素;仅在位置0处。因此,您将获得“列表索引超出范围”。

换句话说,orig_img_names中只有一个图像可供选择。

相关问题