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]))