python按照比例分配数据集：train、test、val

╰半橙微兮° 2023-02-10 05:12 6阅读 0赞

#把所有的xml文件分类为：val train test trainval
    #比例是：6:2:2
    import os
    import random
    trainval_percent = 0.8
    train_percent = 0.6
    xmlfilepath = 'Annotations'  #Annotations地址
    txtsavepath = 'ImageSets/Main' #Main地址
    total_xml = os.listdir(xmlfilepath)
    total_xml.sort()
    print(total_xml)
    num = len(total_xml)
    tv = int(num * trainval_percent)
    trainval=[]
    train=[]
    trainval=[i for i in range(tv)]
    val=[i for i in range(0,tv,4)]
    ftrainval = open('/home/dlut/网ftrainval = open('ImageSets/Main/trainval.txt', 'w')
    ftest = open('ImageSets/Main/test.txt', 'w')
    ftrain = open('ImageSets/Main/train.txt', 'w')
    fval = open('ImageSets/Main/val.txt', 'w')
    for i in range(num):
        name = total_xml[i][:-4] + '\n'
        print(name)
        if i in trainval:
            ftrainval.write(name)
            if i in val:
                # ftrain.write(name)
                fval.write(name)
            else:
                # fval.write(name)
                ftrain.write(name)
        else:
            ftest.write(name)
    ftrainval.close()
    ftrain.close()
    fval.close()
    ftest.close()

发表评论取消回复

表情：

评论列表（有 0 条评论，6人围观）

还没有评论，来说两句吧...

相关阅读

相关介绍一些资源分配里面的比例公平算法

资源分配的比例公平算法是指通过某种方式将资源分配给参与者，使得每个参与者所获得的资源比例相同或基本相同的算法。以下是一些比例公平算法： 1. 等份分配算法：将资源平均分配给

太过爱你忘了你带给我的痛/ 2024年03月25日 13:16/ 0 赞/ 116 阅读

相关 flex比例、顺序分配

<!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Document</

蔚落/ 2023年10月17日 19:08/ 0 赞/ 75 阅读

相关 python按行分割文件：python 划分数据集文件(txt标签文件按比例随机切分)

import os import random class SplitFiles(): """按行分割文件"""

川长思鸟来/ 2023年09月29日 12:53/ 0 赞/ 85 阅读

相关 java将图片按照原尺寸比例存入word中

在使用poi往word里面插入图片时可以使用 XWPFParagraph.insertNewRun(i).addPicture(InputStream picture

你的名字/ 2023年07月06日 08:49/ 0 赞/ 41 阅读

相关 python按照比例分配数据集：train、test、val

把所有的xml文件分类为：val train test trainval 比例是：6:2:2 import os import random

╰半橙微兮°/ 2023年02月10日 05:12/ 0 赞/ 7 阅读

相关 Android布局的高按照比例来分布

线性布局（五比一）： <LinearLayout android:layout_width="match_parent"

一时失言乱红尘/ 2022年07月28日 04:06/ 0 赞/ 198 阅读

相关按照特征值划分数据集

创建数据集： def createDataSet(): dataSet = [[1, 1, 'yes'], [1

女爷i/ 2022年06月04日 06:26/ 0 赞/ 283 阅读

相关 JVM堆老年代分配比例

老年代分配 \-XX:NewRatio 设置老年代与新生代的比例 \-XX:NewRatio=老年代/新生代尽可能将对象，预留在新生代减少老年代的GC次

红太狼/ 2022年05月16日 09:25/ 0 赞/ 298 阅读

相关 JVM堆新生代分配比例

新生代分配 \-XX:SurvivorRatio 设置新生代分配比例表示Eden空间与from/to空间的比例 XX:SurvivorRatio=eden/f

川长思鸟来/ 2022年05月16日 09:24/ 0 赞/ 299 阅读

相关 Python 把标注文件Annotations.txt按7：2：1数量比例分配给train集，val集，test集，其中单个样本随机分配

先测试把100个数字0-99个，按7：2：1数量比例分配给a集，b集，c集，其中单个样本随机分配 import random totalnum =

╰半夏微凉°/ 2021年11月17日 12:04/ 0 赞/ 123 阅读