根据索引文件划分数据集

索引文件.txt:

根据索引文件中的视频编号划分训练集，测试集，验证集

g = open('./ACRN/TACoS/TACoS_val_videos.txt','r')
label=g.readlines()
print(label)
len(label)

 

path = r'.\videos\video'
x = path+"\\"+label[0].strip('\n').split(',')[0]
x

import shutil
path = r'.\videos\video'
path_new = r'.\videos\val_data'
for i in range(len(label)):
    name = label[i].strip('\n').split(',')[0]
    print(name)
    print(path+'\\'+name)
    name_new = label[i].strip('\n').split(',')[0].split('.')[0]
    #num = label[i].strip('\n').split(',')[1]
    root = path_new + '\\'+ name_new + '.avi'
    print(root)
    shutil.copyfile(path+'\\'+name, root)

按照索引文件的编号，将指定数据文件复制到已经建好的文件夹中