python - 如何将所有测试数据图像保存到pytorch Lightning中的文件夹
问题描述
我已经在 PyTorch 闪电中实现了一个 UNet,并且我得到了结果。现在我想将所有测试数据输出保存到与测试图像同名的文件夹中。例如,如果测试图像名称是 image1,那么我需要将测试输出图像也保存为 image1。我已经浏览了 PyTorch 闪电文档,我开始了解
trainer.predict
但我不知道如何实现它。
解决方案
推荐阅读
- python - 在 for 循环中更改段落字体属性会更改所述循环内所有段落的属性
- cygwin - cygwin 是否支持 api int getrusage(int who, struct rusage *usage); 中的 RUSAGE_THREAD?
- spring-boot - 如何在 Spring Tool Suite 中的数据 JPA 的 findBy 方法中获取建议
- django-views - 如何使删除按钮在 Django 中工作
- appearance - 即使在 MDN,CSS 外观也无法正常工作
- java - Web 应用程序中的预测模型
- django - 使用 Django 视图使用 POST 方法将用户重定向到第 3 方应用程序
- django - 如何自动填充来自其他模型的数据以及如何添加计算字段?
- ruby-on-rails - POST 到 https 总是得到 404 错误,但如果使用 http 则不会
- embedded - 无法在 Windows 10 上使用 Visual Studio Code 打开源文件 avr/io.h(hal.h 的依赖项)