制作训练时需要生成txt文件

创作人：ZLY

---

前言

在进行训练的时候，必不可少的需要使用的文件是txt文件，用来进行数据读取

---

一、前期准备

1.库要求

本次使用的库包含：
OS，numpy，python 无版本限制

2.存储格式

1.文件内容展示

2.文件夹展示

3. train.txt文件展示
   
   4.label标签文件展示

二、使用步骤

1.引入库

代码如下（示例）：

import os
import random #乱序使用随机函数

2.设置保存地址

地址为：train，eval，label

#保存地址
first_filedir="/home/aistudio/ttst"
traindir="/home/aistudio/train_set.txt"
evaldir="/home/aistudio/eval_set.txt"

3.判断该地址下，是否含有已经存在的txt

if(os.path.exists(traindir)):  # 判断有误文件
    os.remove(traindir)  # 删除文件
if(os.path.exists(evaldir)):  # 判断有误文件
    os.remove(evaldir)  # 删除文件
fp = open(traindir,'w')
fp = open(evaldir,'w')

4.构建生成train，eval的函数

def make_txt(first_filedir,traindir,evaldir):
    #构建训练列表
    train_list=[]
    eval_list=[]
    filelist = os.listdir(first_filedir) #获取主目录下的子文件如N0，N1
    for first_dir in filelist:
        second_filedir=first_filedir+'/'+first_dir# 遍历到/home/aistudio/ttst/N0
        second_filelist=os.listdir(second_filedir)# 读取主目录下的子文件如N0_NPY
        count = 0#计数
        for second_dir in second_filelist:
            count = count + 1 
            third_filedir=second_filedir+'/'+second_dir#真正的指向了文件N0_0_21.npy文件
            if count % 10 == 0:#定义为每10张图片，取一张为测试图片
                #此处可以自定义*
                eval_list.append(third_filedir + ' ' + second_dir[1] + '\n')
            else:
                train_list.append(third_filedir + ' ' + second_dir[1] + '\n')

6.乱序

    #乱序    
    random.shuffle(eval_list)
    #读写
    with open(evaldir, 'a') as f:
        for eval_image in eval_list:
            f.write(eval_image)
    #乱序    
    random.shuffle(train_list)
    with open(traindir, 'a') as f:
        for train_image in train_list:
            f.write(train_image)

7.制作predict（test）文件

import os
if(os.path.exists('/home/aistudio/test_set.txt')):  # 判断有误文件
    os.remove('/home/aistudio/testing_set.txt')  # 删除文件

fp = open("/home/aistudio/test_set.txt",'w')

def make_train_txt(first_filedir):
    with open(f'/home/aistudio/test_set.txt', 'a') as f_train:
            filelist = os.listdir(first_filedir) #获取主目录下的子文件如N0，N1
            for first_dir in filelist:
                second_filedir=first_filedir+'/'+first_dir# 遍历到/home/aistudio/ttst/N0
                second_filelist=os.listdir(second_filedir)# 读取主目录下的子文件如N0_NPY
                for second_dir in second_filelist: 
                    third_filedir=second_filedir+'/'+second_dir#真正的指向了文件N0_0_21.npy文件
                    f_train.write(third_filedir + '\n')

make_train_txt("/home/aistudio/ttst")

---

总结

本次实验实现了如何制作txt文件，以npy文件为例子

完整代码
import os
import random #乱序使用随机函数
#保存地址
first_filedir="/home/aistudio/ttst"
traindir="/home/aistudio/train_set.txt"
evaldir="/home/aistudio/eval_set.txt"
#判断该地址下，是否含有已经存在的txt
if(os.path.exists(traindir)):  # 判断有误文件
    os.remove(traindir)  # 删除文件
if(os.path.exists(evaldir)):  # 判断有误文件
    os.remove(evaldir)  # 删除文件
fp = open(traindir,'w')
fp = open(evaldir,'w')
#构建生成txt文件函数，传入参数为（文件夹地址，训练txt地址，测试txt地址）
def make_txt(first_filedir,traindir,evaldir):
    #构建训练列表
    train_list=[]
    eval_list=[]
    filelist = os.listdir(first_filedir) #获取主目录下的子文件如N0，N1
    for first_dir in filelist:
        second_filedir=first_filedir+'/'+first_dir# 遍历到/home/aistudio/ttst/N0
        second_filelist=os.listdir(second_filedir)# 读取主目录下的子文件如N0_NPY
        count = 0#计数
        for second_dir in second_filelist:
            count = count + 1 
            third_filedir=second_filedir+'/'+second_dir#真正的指向了文件N0_0_21.npy文件
            if count % 10 == 0:#定义为每10张图片，取一张为测试图片
                #此处可以自定义*
                eval_list.append(third_filedir + ' ' + second_dir[1] + '\n')
            else:
                train_list.append(third_filedir + ' ' + second_dir[1] + '\n')
    #乱序    
    random.shuffle(eval_list)
    #读写
    with open(evaldir, 'a') as f:
        for eval_image in eval_list:
            f.write(eval_image)
    #乱序    
    random.shuffle(train_list)
    with open(traindir, 'a') as f:
        for train_image in train_list:
            f.write(train_image)
#执行函数
make_txt(first_filedir,traindir,evaldir)