如何将ID附加到导入和numpy数组转换的图像?

Question

我正在将图像导入Python并将它们转换为numpy数组以进行相似性分析。每个文件都有一个类型为"32451.png"的名称。我想做的是以某种方式将文件名(在本例中为"32451"(附加到数组中的每个元素。我需要这样做，因为我有一个基于该文件号的训练/测试元素列表，所以我需要能够将它们分开。我目前在下面使用的代码引入了图像并将它们转换为 numpy 数组，但据我所知，它不会将标识符保存在我可以用来关联回图像文件名的任何地方。

imgs=[]
path="Documents/data/img"
os.listdir(path)
valid_images =[".png"]
for f in os.listdir(path):
ext= os.path.splitext(f)[1]
if ext.lower() not in valid_images:
continue
imgs.append(np.asarray(Image.open(os.path.join(path,f))))

Answer 1

感谢您的建议，这是我能够想出的代码。

imgs_dict={}
path="Documents/data/img"
os.listdir(path)
valid_images =[".png"]
for f in os.listdir(path):
ext= os.path.splitext(f)[1]
if ext.lower() not in valid_images:
continue
img_name=os.path.basename(f)
img_name=os.path.splitext(img_name)[0]
img=np.asarray(Image.open(os.path.join(path,f)))
imgs_dict.update([(img_name,img)])