首页 > 解决方案 > 如何将所有测试数据图像保存到pytorch Lightning中的文件夹

问题描述

我已经在 PyTorch 闪电中实现了一个 UNet,并且我得到了结果。现在我想将所有测试数据输出保存到与测试图像同名的文件夹中。例如,如果测试图像名称是 image1,那么我需要将测试输出图像也保存为 image1。我已经浏览了 PyTorch 闪电文档,我开始了解

trainer.predict

但我不知道如何实现它。

标签: pythonpytorchpytorch-lightning

解决方案


推荐阅读