快速生成训练样本与标签样本之间的索引文件

来源:互联网 发布:fedora 24 php 编辑:程序博客网 时间:2024/06/03 03:36

我们经常在训练的时候会用到将训练文件与标签文件对应起来的索引文件,如下格式:
这里写图片描述
这个索引有两部分组成:训练文件+空格+标签文件
本文使用先分别生成训练文件索引、标签文件索引,再将两部分合成的方法,灵活且高效地生成训练样本与标签样本之间的索引文件。

(1)训练与标签索引生成

可以使用matlab文件生成

%批量生成文件名主函数

clc,clear,close all;SrcPath = 'C:\wzz\01_work\04_anotation\21_low_speed_following\images\';SrcFile = dir([SrcPath,'*.png']);iCount = length(SrcFile);fid = fopen('list.txt','wt');for k = 1:iCount    FileName = [SrcPath,SrcFile(k).name];    fprintf(fid,'%s\n',FileName);endfclose(fid);

(2)合成训练与标签索引到一个文件中

先进行下格式转换,否则容易出现如下问题:
这里写图片描述
出现上述问题后,之间对文件进行dos2unix是不生效的,因为该命令处理的是行尾部,需要像上述分别处理才行。

dos2unix imgs.txtdos2unix label.txt paste -d ' ' imgs.txt label.txt > train.txt

于是,生成了最终需要的训练样本与标签样本之间的索引文件。

这里写图片描述

0 0
原创粉丝点击