概述
最近在做和视觉深度学习有关的项目,在写相关代码的时候读取数据集的图片来进行预处理成了大问题。在查询了网上现有方法的基础上,总结了一种较为简单通用的读取方法。
遍历文件夹
首先是对于数据集文件夹的遍历,这里可以输出得到文件夹内所有图片的名称
def load_img(train_imgPath,test_imgPath):
test_filenames = os.listdir(test_imgPath)
train_filenames = os.listdir(train_imgPath)
for tfname in train_filenames:
if os.path.isdir(tfname) == False:
filenames = os.listdir(