标签 glob 下的文章

python中使用glob模块获得图像路径

python中glob模块可以获得图像路径，目录中的文件名分别为：1.jpg，2.jpg，3.jpg，4.png，5.png。

使用方法为：

1
2
3
4
5
6
7
import glob
image_path_list = glob.glob(r"train/*")
image_path_list.sort()
print (image_path_list)
jpg_path_list = glob.glob(r"train/*.jpg")
jpg_path_list.sort()
print (jpg_path_list)

输出：

1
2
['train/1.jpg', 'train/2.jpg', 'train/3.jpg', 'train/4.png', 'train/5.png']
['train/1.jpg', 'train/2.jpg', 'train/3.jpg']

对于使用mask-rcnn的数据集，目录下为多个以图像名称为名的文件夹，每个文件夹中有image和masks文件夹，其中image文件夹中有一张图像，表示原图，masks文件夹中有多个与原图大小相同的图像，表示图像中的各个目标。

读取方法为：

1
2
3
4
5
6
7
8
9
10
import glob
import os
image_list = glob.glob(r"dataset/*")
image_list.sort()
print (image_list)
for image_path in image_list:
    image = glob.glob(os.path.join(image_path,'image','*'))[0]
    print (image)
    mask_list = glob.glob(os.path.join(image_path,'masks','*'))
    print (mask_list)

输出结果为：

1
2
3
4
5
['dataset/1', 'dataset/2']
dataset/1/image/1.jpg
['dataset/1/masks/1_1.jpg', 'dataset/1/masks/1_2.jpg', 'dataset/1/masks/1_0.jpg']
dataset/2/image/2.jpg
['dataset/2/masks/2_0.jpg', 'dataset/2/masks/2_1.jpg']