python中使用glob模块获得图像路径
python中glob模块可以获得图像路径,目录中的文件名分别为:1.jpg,2.jpg,3.jpg,4.png,5.png。
使用方法为:
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)
输出:
['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文件夹中有多个与原图大小相同的图像,表示图像中的各个目标。
读取方法为:
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)
输出结果为:
['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']