按占比划分文件，并将文件名写入txt文件(train、valid、trainvalid、test)

代码如下：

import os
import random
import argparseparser = argparse.ArgumentParser()
# 1.xml文件的文件夹路径，根据自己的数据进行修改。 xml一般存放在Annotations下
parser.add_argument('--xml_path', default='D:/数据集/dataset/SeaShips(7000)/Annotations', type=str, help='input xml label path')
# 2.保存 数据集划分生成的txt文件 的文件夹路径。
parser.add_argument('--txt_path', default='D:/数据集/dataset/SeaShips(7000)/ImageSets/Main', type=str, help='output txt label path')
opt = parser.parse_args()trainval_percent = 1  # 3.训练+验证集一共所占的比例，剩下的就是测试集了。
train_percent = 0.9     # 4.训练集在训练集和验证集总集合中占的比例
xmlfilepath = opt.xml_path
txtsavepath = opt.txt_path
total_xml = os.listdir(xmlfilepath)
if not os.path.exists(txtsavepath):os.makedirs(txtsavepath)num = len(total_xml)
list_index = range(num)
tv = int(num * trainval_percent)
tr = int(tv * train_percent)
trainval = random.sample(list_index, tv)
train = random.sample(trainval, tr)file_trainval = open(txtsavepath + '/trainval.txt', 'w')
file_test = open(txtsavepath + '/test.txt', 'w')
file_train = open(txtsavepath + '/train.txt', 'w')
file_val = open(txtsavepath + '/val.txt', 'w')for i in list_index:name = total_xml[i][:-4] + '\n' # 5.-4可以用来去掉文件后缀名if i in trainval:file_trainval.write(name)if i in train:file_train.write(name)else:file_val.write(name)else:file_test.write(name)file_trainval.close()
file_train.close()
file_val.close()
file_test.close()

按占比划分文件，并将文件名写入txt文件(train、valid、trainvalid、test)相关推荐

java实现读取txt文件和将内容写入txt文件
1.java读取txt和写入txt文件 package com.ztesoft.TestClusterHHHT;import java.io.BufferedReader; import java.i ...
按占比划分文件，并将文件名写入txt文件
全部文件: 生成的文件: 注:文件名可自行在代码中决定加不加代码: import os import random import argparseparser = argparse.Argument ...
划分数据集代码（按照4:1的比例）以及根据各自文件名写入txt文件
会将图片分到两个文件夹中: #include <opencv2/opencv.hpp> #include "opencv2/features2d.hpp" #inclu ...
python 写文件换行_python中写入txt文件需要换行,以及\r 和\n
原博文 2019-09-25 16:16 − 在Python中,用open()函数打开一个txt文件,写入一行数据之后需要一个换行如果直接用 f.write('\n')只会在后面打印一个字符串'\n ...
java追加txt文件_java怎么追加写入txt文件
java中,对文件进行追加内容操作的三种方法! import java.io.BufferedWriter;import java.io.FileOutputStream;import java.io ...
java 导入导出txt文件_Java读取和写入txt文件
1 问题描述对于java的读取和写入txt一直心存疑惑,随着知识的积累,又重新进行学习,对java的文件读写理解更加深刻,在这里将自己的小小经验总结分享给大家.下面是大家了解java流的一个基本框架 ...
java 追加写入txt文件_ava如何追加写入txt文件
(二)方法1 public void method1() { FileWriter fw = null; try { //如果文件存在,则追加内容:如果文件不存在,则创建文件 File f=new F ...
matalb读取txt文件以及将数据写入txt文件
matlab读取txt文件三个文件a.txt,b.txt,d.txt 1.load load只能读取ascii文件,如果文件中有不同数据类型将出错. 2.importdata 3.textread ...
android 读写文件简书,Android写入txt文件并读取
直接上代码~ 注意权限 1.写入数据 private void writeData() { String filePath = "/sdcard/Gyt/"; String fil ...

按占比划分文件，并将文件名写入txt文件(train、valid、trainvalid、test)

按占比划分文件，并将文件名写入txt文件(train、valid、trainvalid、test)相关推荐

最新文章

热门文章