Python：将图片和关联的文件夹名称存储在列表中

Question

我对Python非常陌生，我想知道是否有人可以帮助我解决这个问题。在

我有一个文件夹“火车”与子文件夹“球”，“椅子”，“杯子”，“袋子”每个子文件夹都由它组成关联像这样作为：

project/Train/ball/01.jpg
project/Train/chair/02.jpg
project/Train/cup/03.jpg
project/Train/bag/04.jpg

我写了以下代码从火车上读取图像

^{pr2}$

对于names_path = []我想存储当前图像的子文件夹（bag、chat、cup、ball）的名称。因此，如果循环中的第一个图像属于ball，则将图像路径保存到training_paths，并将“ball”保存到names_path

我尝试过使用另一个循环，例如

for clas in glob.glob('/Train/*/'):

但是当我检查列表时，所有的条目都没有。在

Answer 1

如果file_path是图像的路径，则可以通过以下方式获取目录名：

from os import path
file_path = "project/Train/ball/01.jpg"
dir_name = path.basename(path.dirname(file_path))  # "ball"

• dirname()提取目录的路径
• basename()提取目录路径的最后一部分
• 另请参见：https://docs.python.org/3/library/os.path.html

Answer 2

这将取决于您实际从何处运行该文件，但我认为您的错误是您不需要在Train之前使用斜杠。在

更改：

glob.glob('/Train/**/*.jpg')

收件人：

^{pr2}$

您的代码正在生成一个空列表，因此for循环永远不会运行。在