tool_Data_To_Train_And_CV_data.py

# -*- coding: utf-8 -*-
"""
Created on Mon Feb 25 08:49:49 2019

@author: Dufert
"""
import os
from os import listdir, getcwd
from os.path import join
file_num = 0
if __name__ == '__main__':
    source_folder='g:/CV_Library/Winding_data/winding3/train/imgs/'#地址是所有图片的保存地点
    dest='g:/CV_Library/Winding_data/winding3/train/train.txt' #保存train.txt的地址
    dest2='g:/CV_Library/Winding_data/winding3/train/val.txt'  #保存val.txt的地址
    file_list=os.listdir(source_folder)       #赋值图片所在文件夹的文件列表
    train_file=open(dest,'a')                 #打开文件
    val_file=open(dest2,'a')                  #打开文件
    for file_obj in file_list:                #访问文件列表中的每一个文件
        file_path=os.path.join(source_folder,file_obj) 
        #file_path保存每一个文件的完整路径
        file_name,file_extend=os.path.splitext(file_obj)
        #file_name 保存文件的名字，file_extend保存文件扩展名
#        file_num=int(file_name) 
        file_num += 1
        #把每一个文件命str转换为 数字 int型 每一文件名字都是由四位数字组成的  如 0201 代表 201     高位补零  
        if(file_num<200):                     #保留900个文件用于训练
            #print file_num
            train_file.write(file_name+'\n')  #用于训练前900个的图片路径保存在train.txt里面，结尾加回车换行
        else :
            val_file.write(file_name+'\n')    #其余的文件保存在val.txt里面
    train_file.close()#关闭文件
    val_file.close()