Question

我有一个目录（human_image_dir），其中包含.jpg格式的图像。我想遍历每张图像，并将图像文件的名称添加为字典的键，并将分数作为值。但是我不确定我编写的代码。多谢您的协助。我尝试了以下代码：

score_dict = {}

obj1_feature = feature.calcFeature(current_human_image)
for image in self.human_image_dir:
      obj2_feature = feature.calcFeature(image)
      score = np.score = np.linalg.norm(obj1_feature - obj2_feature)
      if score < self.threshold:
            score_dict["key"] = "image"
            score_dict["value"] = score

预期输出：

score_dict = {"image": score }

Answer 1

您可以使用os列出目录中的所有文件，然后在将新值插入字典中时，只需使用os中的名称作为键即可。

您还需要将文件读入内存，您可以在其中使用matplotlib.image

import os, sys
import matplotlib.image as mpimg

score_dict = {}

for image in os.listdir(self.human_image_dir):
    img = mpimg.imread(os.path.join(self.human_image_dir,image))
    obj2_feature = feature.calcFeature(img)
    score = np.score = np.linalg.norm(obj1_feature - obj2_feature)
    if score < self.threshold:
        score_dict[image] = score

如何将目录中的文件名保存为字典的键？

1 个答案: