如何在python中打开并选择.rec文件中的图像子集?

2024-09-22 16:37:39 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个包含以下文件的文件夹:火车.idx, 火车.rec,财产,垃圾桶,cfp_fp.bin,agedb_30.bin

此文件夹包含人脸图像。在

我已经在下面的代码中使用了ImageRecordIter。在

然而,每次打印train_数据的第一个元素时,我都会得到不同的图像。在

train_data = ImageRecordIter(
    path_imgrec = os.path.join(rec_path,'train.rec'),
    path_imgidx = os.path.join(rec_path, 'train.idx'),
    label_width = 2,

    data_shape  = (3,112,112 ),
    batch_size  = 10,
    shuffle     = False)

我的问题是:

1)我不知道数据通常如何存储在这些类型的文件中,例如哪种类型的文件包含标签。你知道这些类型的文件吗?在

2)如何提取数据子集以制作示例文件?另外,文件格式是什么(例如pickle文件、txt文件)?在


Tags: 文件数据path图像文件夹类型databin
1条回答
网友
1楼 · 发布于 2024-09-22 16:37:39

您只在ImageRecordIter中使用train.rec和{},因此它们是唯一被使用的文件。您的标签将存储在train.rec文件中(与数据一起)。您可以使用^{}从这些文件中提取随机样本。比如:

samples = []
record = mx.recordio.MXIndexedRecordIO('tmp.idx', 'tmp.rec', 'r')
for i in range(5):
    samples.append(record.read_idx(i))

record = mx.recordio.MXIndexedRecordIO('tmp.idx', 'tmp.rec', 'w')
for i, sample in enumerate(samples):
    record.write_idx(i, sample)
record.close()

相关问题 更多 >