File size: 341 Bytes
8cf4db8
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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')