deep-image-prior 调试笔记

GITHUB:https://github.com/DmitryUlyanov/deep-image-prior

使用 jupyter

执行   jupyter notebook 进入

默认只支持本地,网络访问参考  图片的语义分割实战

执行哪个文件都报错(貌似  3.5 才报错  2.7 没事)

 No module named 'skip'

修改  ~/models/__init__.py  前几行

from models.skip import skip
from models.texture_nets import get_texture_nets
from models.resnet import ResNet
from models.unet import UNet

修改 ~/models/common.py

from downsampler import Downsampler -->  from models.downsampler import Downsampler

出错的地方有点多,反正找不到就加 models 吧  /(ㄒoㄒ)/~~

例外:

~/utils/denoising_utils.py

from common_utils import *   -->   from utils.common_utils import *

 

报错

float() argument must be a string or a number, not '_ImageCrop'

img_np 进行如下修改

img_np = get_image(fname, imsize)[1]

 

RUN 成功,效果还不错

这套东西在  python3.5 上坑实在太多,建议还是 2.7 下跑。

重新布了一个 2.7 的环境,吐槽一下  pytorch 直接pip 在线安装太慢了。下载下来再装快的多

生成混淆图
2981轮迭代
2499轮迭代
1987轮迭代
1475轮迭代
993轮迭代
481轮迭代

《deep-image-prior 调试笔记》上有20条评论

          1. 他这个代码的超分辨率部分博主看懂了嘛,,,怎么生成的并不清晰呢。。难道不是在原基础上提高分辨率的??

          1. 我理解,因为这个网络没有经过预训练,所以它超分重建应该是基于一个更低像素的图片,然后慢慢逼近原图,这个结果未必能达到超分重建的目标。

  1. 博主我碰到两个bug
    cuda runtime error (38) : no CUDA-capable device is detected at ..\aten\src\THC\THCGeneral.cpp:51
    [Errno 2] No such file or directory: ‘alexnet-torch_py3.pth’
    请问是不是我的pytorch有问题啊

发表评论

您的电子邮箱地址不会被公开。

此站点使用Akismet来减少垃圾评论。了解我们如何处理您的评论数据