#!/usr/bin/env python
# -*- coding:utf-8 -*-
# 将一个文件夹下图片保存在txt文本
import sys
sys.path.append('D:\\train\\img\\image_valid') (图片文件夹)
import os
import random
img_dir = 'D:\\train\\img\\image_valid'
file_list = []
write_dir_name = 'D:\\train\\img\\img_valid.txt' (图片文件名,不含jpg)
write_file = open(write_dir_name, "w")
for file in os.listdir(img_dir):
## if file.endswith(".jpg"):
write_name = file
write_name1 = write_name.split('.')
print(write_name1)
write_name2 = write_name1[0]
print(write_name2)
file_list.append(write_name2)
sorted(file_list)
num = len(file_list)
for current_line in range(num):
write_file.write(file_list[current_line] + '\n')
write_file.close()