[已解决]为什么会报错，怎么解决呢

yangbaowen

1. import os
   
2. 
   
3. train_txt_path = os.path.join('data', 'catVSdog', 'train.txt')
   
4. train_dir = os.path.join("data", "catVSdog", "train_data")
   
5. valid_txt_path = os.path.join("data", "catVSdog", "test.txt")
   
6. valid_dir = os.path.join("data", "catVSdog", "test_data")
   
7. 
   
8. def gen_txt(txt_path, img_dir):
   
9.     f = open(txt_path, 'w')
   
10.    
    
11.     for root, s_dirs, _ in os.walk(img_dir, topdown=True):  # 获取 train文件下各文件夹名称
    
12.         for sub_dir in s_dirs:
    
13.             i_dir = os.path.join(root, sub_dir)             # 获取各类的文件夹 绝对路径
    
14.             img_list = os.listdir(i_dir)                    # 获取类别文件夹下所有png图片的路径
    
15.             for i in range(len(img_list)):
    
16.                 if not img_list[i].endswith('jpg'):         # 若不是png文件，跳过
    
17.                     continue
    
18.                 #label = (img_list[i].split('.')[0] == 'cat')? 0 : 1
    
19.                 label = img_list[i].split('.')[0]
    
20.                 # 将字符类别转为整型类型表示
    
21.                 if label == 'cat':
    
22.                     label = '0'
    
23.                 else:
    
24.                     label = '1'
    
25.                 img_path = os.path.join(i_dir, img_list[i])
    
26.                 line = img_path + ' ' + label + '\n'
    
27.                 f.write(line)
    
28.     f.close()
    
29. 
    
30. if __name__ == '__main__':
    
31.     gen_txt(train_txt_path, train_dir)
    
32.     gen_txt(valid_txt_path, valid_dir)

复制代码

最佳答案

月排行榜 / 总排行榜

jackz007

2022-10-27 15:15:16

        不知道你到底想要干什么，我就权当是在指定的目录及其子目录内寻找所有的图片文件，并记录在指定文本文件中。程序线索非常清晰，你应该有能力自己按需要改的。

1. import os
   
2. 
   
3. def gen_txt(fn , img_dir):
   
4.     f = open(fn , 'w')
   
5.     for root , dirs , files in os . walk(img_dir):
   
6.         for each in files:
   
7.             x = os . path . join(root , each)
   
8.             ext = os . path . splitext(each)[1] . lower()
   
9.             if ext in ['.jpg' , '.jpeg' , '.png' , '.gif']:
   
10.                 f . write(x + '\n')
    
11.     f . close()
    
12. 
    
13. if __name__ == '__main__':
    
14.     train_txt_path = os.path.join('data', 'catVSdog', 'train.txt')
    
15.     train_dir = os.path.join("data", "catVSdog", "train_data")
    
16.     valid_txt_path = os.path.join("data", "catVSdog", "test.txt")
    
17.     valid_dir = os.path.join("data", "catVSdog", "test_data")
    
18. 
    
19.     gen_txt('train.txt' , train_dir)
    
20.     gen_txt('test.txt' , valid_dir)

复制代码

        只要用了 os . walk() 再用 os . listdir() 那就纯粹多余了，完全没有必要，因为，前者足以搞定目录树上任何一层目录内的任何一个文件。

跳转到最佳答案楼层

jackz007

        不知道你到底想要干什么，我就权当是在指定的目录及其子目录内寻找所有的图片文件，并记录在指定文本文件中。程序线索非常清晰，你应该有能力自己按需要改的。

1. import os
   
2. 
   
3. def gen_txt(fn , img_dir):
   
4.     f = open(fn , 'w')
   
5.     for root , dirs , files in os . walk(img_dir):
   
6.         for each in files:
   
7.             x = os . path . join(root , each)
   
8.             ext = os . path . splitext(each)[1] . lower()
   
9.             if ext in ['.jpg' , '.jpeg' , '.png' , '.gif']:
   
10.                 f . write(x + '\n')
    
11.     f . close()
    
12. 
    
13. if __name__ == '__main__':
    
14.     train_txt_path = os.path.join('data', 'catVSdog', 'train.txt')
    
15.     train_dir = os.path.join("data", "catVSdog", "train_data")
    
16.     valid_txt_path = os.path.join("data", "catVSdog", "test.txt")
    
17.     valid_dir = os.path.join("data", "catVSdog", "test_data")
    
18. 
    
19.     gen_txt('train.txt' , train_dir)
    
20.     gen_txt('test.txt' , valid_dir)

复制代码

        只要用了 os . walk() 再用 os . listdir() 那就纯粹多余了，完全没有必要，因为，前者足以搞定目录树上任何一层目录内的任何一个文件。

阿奇_o

路径，路径，路径！！！

kishere

你给的这路径对吗。。。你这windows平台不得用相对路径，要不就带盘符，你这写的两个都不是