数据集读取名字并放入txt

import numpy as np
import os 
import cv2

from PIL import Image
from glob import glob
# def readimg(path):
#     img_list = []
#     for i in os.listdir(path):
#         name = os.path.join(i)
#         img_list.append(name)
#     return img_list

files = glob('/home/data/*/*.jpg')
print(len(files))
# print(np.array(files).shape)
# for files in files:

# # 2.划分train与valid~k折
# K = 5
# files = glob('/home/data/*/*.txt')
# random.shuffle(files)
# ind = len(files)//K
ind = int((len(files)*4)/5)
# print(ind)
# train = [x.replace('.txt','.jpg') for x in files[:ind]]
# valid = [x.replace('.txt','.jpg') for x in files[ind:]]
train = [x for x in files[:ind]]
valid = [x for x in files[ind:]]
print(f"train{ind},valid{int(len(valid)/5)}")

# # 3.写入文件
with open('/project/train/src_repo/data/train.txt','w') as f:
    f.write('\n'.join(train))
with open('/project/train/src_repo/data/valid.txt','w') as f:
    f.write('\n'.join(valid))
with open('/project/train/src_repo/train.txt','w') as f:
    f.write('\n'.join(train))
with open('/project/train/src_repo/valid.txt','w') as f:
    f.write('\n'.join(valid))
print(files)