利用python读取文件目录和图片

读取文件主要用到os，如果读取的是图片信息，还需要利用cv框架，对于以后目标检测也有很大用处
这里直接上代码

导入模块

import os
import cv2

定义字符串的后缀与标签匹配函数

#输入一个字符串一个标签，对这个字符串的后续和标签进行匹配
def endwith(s,*endstring):
    resultArray = map(s.endswith,endstring)
    if True in resultArray:
        return True
    else:
        return False

s这里代表列表中每个图片的文件名字符串
*endstring代表各种后缀字符串，例如“.jpg、.png"等图片格式
主要通过map函数进行匹配
后面条件语句用于判断文件中是否有指定的图片内容
retrun true； 返回正确的处理结果。

return false；返回错误的处理结果；终止处理；阻止提交表单；阻止执行默认的行为（只对本函数有用）。

return；把控制权返回给页面。

读取指定文件夹图片

#根据输入的文件夹绝对路径，将该文件夹下的所有指定suffix的文件读取存入一个list,该list的第一个元素是该文件夹的名字
def readAllImg(path,*suffix):
    try:

        s = os.listdir(path)
        resultArray = []
        #添加文件名
        fileName = os.path.basename(path)
        resultArray.append(fileName)
		#添加图片
        for i in s:
            if endwith(i, *suffix):
                document = os.path.join(path, i)
                img = cv2.imread(document)
                resultArray.append(img)
                
    except IOError:
        print("读取失败！")

    else:
        print("读取成功！")
        return resultArray

可视化图片

if __name__ == '__main__':

    result = readAllImg("ch09\shuju\LeiFeng_",'.jpg')
    print (result[0])
    cv2.namedWindow("Image")
    cv2.imshow("Image", result[1])
    cv2.waitKey(0)
    cv2.destroyAllWindows()