训练、测试和标记单独的文件夹

Question

我正在处理计算机视觉任务，我有一个测试、训练和标签文件夹。标签文件夹具有数字标签 1 到 7，它们与训练和测试文件夹中的不同图像对齐。我已经设法用下面的代码绘制了 10 个测试图像（作为测试）：

fig, axes = plt.subplots(nrows=2, ncols=5, figsize=(10, 5), sharex=True, sharey=True)
ax = axes.ravel()

for i in range(len(test_images)):
    ax[i].imshow(test_images[i], cmap='gray')
    
    ax[i].set_axis_off()
fig.tight_layout()
plt.show()

上面的代码仅返回图像的绘图。我打算将相关标签作为标题。标签文件夹包含单个 .txt 标签文件。我需要阅读帮助，然后将标签设置为测试集中相应图像的标题。