我已将我的签名文件夹拆分为训练,验证和测试。现在,我想看到随机选择了两个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
答案 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中只有一个图像可供选择。