Python 提取文件中的图像，并将文件名作为标签

在学习机器学习问题时，我有一个问题。我有一个名为Real World的文件，里面有60个名为clocks、Bed、Bike的文件。所有这60个文件都包含这类图像。因此，我想将所有这些图像输入python（可能是一个NumPy数组），以分析并将文件名（如“clock”）作为该文件中图像的标签。有办法吗

您可以使用内置

os

库中的函数来执行此操作。 下面的代码以NumPy数组列表的形式提供图像，并在列表中显示标签名称

 import os
dir="/dir"
classes=[clas.name for clas in os.scandir(dir) if clas.is_dir()]

X,y=[],[]
for clas in classes:
    path=os.path.join(dir,clas)
    for img in os.listdir(path):
        X.append(plt.imread(os.path.join(path,img)))
        y.append(str(clas))

---

您好，MH563，请发布一些关于这个问题的最简单的例子，到目前为止您尝试过的事情和更详细的信息：包含不同对象图像的文件类型是什么？您希望以后能够对图像执行哪些操作？谢谢。这就像有60个类别（文件名），每个类别包含几个图像（.jpg）。我想训练一个分类器，也许是CNN，来区分这些图片。我需要一些建议来知道如何开始解决这个问题。没有更多关于文件类型的详细信息，我无法详细帮助您。也许这有帮助：对不起。实际上，我指的是文件夹，不是文件。我正在研究的数据集是它是否有用。非常感谢。但是我得到了一个语法错误，

class.name

是无效语法。你知道为什么吗？谢谢。哦，对不起，因为类是一个保留关键字。把名字改成别的名字，它就工作了，这次测试了：D还有一个is_dir的输入错误，我已经用工作代码编辑了答案。