python检索目录下的文件和指定文件

import os def all_files_under(path, extension=None, append_path=True, sort=True): if append_path: if extension is None: filenames = [os.path.join(path, fname) for fname in os.listdir(path)] else: filenames = [os.path.join(path, fname) for fname in os.listdir(path) if fname.endswith(extension)] else: if extension is None: filenames = [os.path.basename(fname) for fname in os.listdir(path)] else: filenames = [os.path.basename(fname) for fname in os.listdir(path) if fname.endswith(extension)] if sort: filenames = sorted(filenames) return filenames result_dir='test/results' datasets=all_files_under(result_dir) for dataset in datasets: imglist = glob.glob(os.path.join('test/label/','*png')) result_dir='test/results' datasets=all_files_under(result_dir) for dataset in datasets: MIOU_pred = [] MDice_pred = [] MAP_pred = [] for i in range(len(imglist)): cls = misc.imread(imglist[i]) pred = misc.imread(dataset+'/'+os.path.basename(imglist[i]))