建立测试集

从数据集中 每个类别等比例 抽取 出图像 建立一个测试集

# -*- coding: utf-8 -*-
"""
Created on Mon Jun 17 17:51:18 2019
从TTKLight 中 提取出 测试集， 分割成两个数据集
这里只获取文件名， 然后使用 bash 命令进行 真正的 move
@author: haithink
"""

imgPath = "E:\\Dataset\\LightFileList.txt"

file = open(imgPath)

iter_f = iter(file)

nameDic = {}

for line in iter_f:
    row = line.split("_")
    #print(row[1])
    
    nameDic.setdefault(row[1], []).append(line[0:-1])

for key, val in nameDic.items():
    count = len(val)
    for i in range(int(count*0.1)):
        print(val[i])