大家好,又见面了,我是你们的朋友全栈君。
由于经常需要遍历某个文件夹及其子文件夹中的图像,因此,写了个精炼的脚本专门用于执行此类操作。
具体如下:
import os
import cv2
def getFileList(dir,Filelist, ext=None):
"""
获取文件夹及其子文件夹中文件列表
输入 dir:文件夹根目录
输入 ext: 扩展名
返回: 文件路径列表
"""
newDir = dir
if os.path.isfile(dir):
if ext is None:
Filelist.append(dir)
else:
if ext in dir[-3:]:
Filelist.append(dir)
elif os.path.isdir(dir):
for s in os.listdir(dir):
newDir=os.path.join(dir,s)
getFileList(newDir, Filelist, ext)
return Filelist
org_img_folder='./org'
# 检索文件
imglist = getFileList(org_img_folder, [], 'jpg')
print('本次执行检索到 '+str(len(imglist))+' 张图像\n')
for imgpath in imglist:
imgname= os.path.splitext(os.path.basename(imgpath))[0]
img = cv2.imread(imgpath, cv2.IMREAD_COLOR)
# 对每幅图像执行相关操作
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/144125.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...