import os def genreate_list(root, split): list_path = os.path.join(root, split+'.txt') with open(list_path, 'w') as f: for img_name in os.listdir(os.path.join(root)): f.write(img_name + '\n') if __name__ == "__main__": root = r'E:\cddataset\mmcd\Second_my\val\im1' genreate_list(root, 'val')