前言

首先,我失败了,我并没有跑通这个代码。一下是一些过程以及猜想。有那位大佬成功了,希望能教教我😂

下面是我自己整理的requirement.txt

numpy
pillow
tensorboardX
torchvision
h5py
pytorch-metric-learning
medpy

一. python freq_space_interpolation_demo.py

会用demo_samples的第四张图像为基准,和前三张图像重构(辅以不同值的$\lambda$),形成新样本。

二. 下载该文章的数据集

Baidu Research Open-Access Dataset - Download

图象是jpg的,label是bmp的,需要做转换,具体的看我的另一篇:(这里的转换是不是

prepare_dataset.py,我也不确定,因为这个代码有问题,参数对不上。)

bmp图像,jpg图像转npy_博弈望天的博客-CSDN博客

三.修改代码:train

3.1直接运行的话,如果报了这个错

ValueError: num_samples should be a positive integer value, but got num_samples=0

可以在第173行(也就是对应的位置),把shuffle的值由true改为false

解释:因为已经有batch_size了,就不需要shuffle来进行随机了,将shuffle设置为FALSE即可

3.2RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.

我的问题是torch没下载全,一些依赖包没有,可以重新下

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐