如何打印在python中for循环的每次迭代中读取的图像名称

Question

我试图在名为“train”的文件夹中找到所有灰度图像。 为此，我的代码是

for img in glob.glob("Tensorflow/workspace/images/train/*.jpg"):
    image = imread(img)
    if(len(image.shape) < 3):
        print('gray')
        # print(image)
        # print(os.path.dirname(os.path.abspath(__file__)))
        # print(__file__)
        # print(os.path.basename('image'))
    elif len(image.shape) == 3:
        print('Color(RGB)')
    else:
        print('others')

我已经尝试了很多方法来做到这一点（我的代码中的注释部分），但是它们打印了 python 文件的路径，例如e:\Desktop\RealTimeFaceRecognition---Microsoft-Engage\Tensorflow\scripts\rgb_checker.py但是我想在每次灰度时打印图像名称，例如，如果发现图像“ mask43 ”是灰度的，那么我想打印E:\Desktop\RealTimeFaceRecognition---Microsoft-Engage\Tensorflow\workspace\images\train\mask43 ，这是图像的路径，而不是我正在检查的 python 文件。

请告诉我这样做所需的代码。

Answer 1

你有一个 for 循环，它在 for 循环中有一个变量img来跟踪它是哪个文件。 这意味着您可以通过print(os.path.dirname(img))访问列表中的当前文件