如何在测试数据集中随机选择人脸

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了如何在测试数据集中随机选择人脸相关的知识,希望对你有一定的参考价值。

我正在使用pytorch和mtcnn进行人脸识别项目,并且在训练了我的训练数据集之后,现在我想对测试数据集进行预测

这是我训练有素的代码

writer = SummaryWriter()
writer.iteration, writer.interval = 0, 10

resnet.eval()
training.pass_epoch(
   resnet, loss_fn, val_loader,
   batch_metrics=metrics, show_running=True, device=device,
   writer=writer
)

for epoch in range(epochs):
    print('
Epoch {}/{}'.format(epoch + 1, epochs))

    resnet.train()
    training.pass_epoch(
        resnet, loss_fn, train_loader, optimizer, scheduler,
        batch_metrics=metrics, show_running=True, device=device,
        writer=writer
     )

    resnet.eval()
    training.pass_epoch(
       resnet, loss_fn, val_loader,
       batch_metrics=metrics, show_running=True, device=device,
       writer=writer
    )

writer.close()

如何在pytorch中进行预测(从测试数据集中随机选择?)>

我正在使用pytorch和mtcnn进行人脸识别项目,并在训练了我的训练数据集之后,现在我想对此测试数据集进行预测,这是我训练有素的代码编写者= SummaryWriter()...

答案
我不确定测试数据的格式,但是要从数据集中随机选择一个样本,可以使用模块random.choice中的random
另一答案
您必须将Dataloader与

shuffle = True

以上是关于如何在测试数据集中随机选择人脸的主要内容,如果未能解决你的问题,请参考以下文章

如何从Matlab中的数据集中选择随机样本[重复]

如何从 Erlang/Elixir 中的 ets 集中选择一个随机元素?

在 Python 多处理进程中运行较慢的 OpenCV 代码片段

如何利用python将txt文件划分训练集和测试集

如何在片段着色器中平铺部分纹理

正确拆分数据集