白云岛资源网 Design By www.pvray.com
如图所示,有一个test文件夹,里面有3个子文件夹,每个子文件夹中有若干图片文件
#场景1 读取一个文件夹中所有文件,存入到一个list表中
#coding:utf-8 from __future__ import absolute_import from __future__ import division from __future__ import print_function import numpy as np import os import sys import math import numpy import time import argparse import random import cv2 def findAllfile(path, allfile): filelist = os.listdir(path) for filename in filelist: filepath = os.path.join(path, filename) if os.path.isdir(filepath): #print(filepath) findAllfile(filepath, allfile) else: allfile.append(filepath) return allfile #场景1 读取一个文件夹中所有文件,存入到一个list表中 def process1(srcpath, imgprocess_result): #遍历图像文件夹 image_files = findAllfile(srcpath,[]) #image_files为所有文件的list #判断 目录是否存在,存在就删除,并且重建 if os.path.exists(imgprocess_result): os.system("rm -rf " + imgprocess_result) if not os.path.isdir(imgprocess_result): # Create the log directory if it doesn't exist os.makedirs(imgprocess_result) #是否随机打乱文件顺序 #random.shuffle(image_files) #遍历所有 for facepath in image_files: print("原始文件路径:", facepath) #获取文件名称 data_split = facepath.strip().split("/") image_floder = data_split[-2] print("文件所在目录:", image_floder) image_name = data_split[-1] print("文件名称:", image_name) image_newfloder = imgprocess_result + "/" + image_floder #判断 目录是否存在,不存在就重建 if not os.path.isdir(image_newfloder): # Create the log directory if it doesn't exist os.makedirs(image_newfloder) image_newpath = image_newfloder + "/" + image_name print("处理后的文件路径:", image_newpath) #开始处理文件 #.............. # # #场景2 首先读取一个文件夹中的所有子目录,然后依次遍历各个子目录的所有文件 def process2(srcpath, imgprocess_result): #找出所有的子目录 filelist = os.listdir(srcpath) for filename in filelist: filepath = os.path.join(srcpath, filename) if os.path.isdir(filepath): print("原始子目录路径:", filepath) image_files = findAllfile(filepath,[]) for facepath in image_files: print("原始文件路径:", facepath) #获取文件名称 data_split = facepath.strip().split("/") image_floder = data_split[-2] print("文件所在目录:", image_floder) image_name = data_split[-1] print("文件名称:", image_name) image_newfloder = imgprocess_result + "/" + image_floder #判断 目录是否存在,不存在就重建 if not os.path.isdir(image_newfloder): # Create the log directory if it doesn't exist os.makedirs(image_newfloder) image_newpath = image_newfloder + "/" + image_name print("处理后的文件路径:", image_newpath) #开始处理文件 #.............. # # if __name__ == '__main__': #原始文件夹 srcpath = "/DATA/share/publicdata/face/dmt_face/test" #处理完毕后存放文件 imgprocess_result = "/DATA/share/publicdata/face/dmt_face/imgprocess_result" print("方法1\n\n\n\n") process1(srcpath, imgprocess_result) print("\n\n\n方法2") process2(srcpath, imgprocess_result)
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。
白云岛资源网 Design By www.pvray.com
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
白云岛资源网 Design By www.pvray.com
暂无评论...