search

zsyOAOA / ResShift

ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shifting (NeurIPS@2023 Spotlight, TPAMI@2024)
Other
870 stars 47 forks source link

找不到 from datapipe.masks.train import process_mask #73

Open yunjiefeng123 opened 3 months ago

yunjiefeng123 commented 3 months ago

尊敬的作者，请问这个train.py在哪里，运行会报错

zsyOAOA commented 3 months ago

能把报错信息写的再详细一点吗？ @yunjiefeng123

yunjiefeng123 commented 3 months ago

就在models文件夹下有一个solvers.py,它的第118行代码有一个 from datapipe.masks.train import process_mask 但是masks是一个.py文件，并没有这个train.py的文件。运行显示 from datapipe.masks.train import process_mask ModuleNotFoundError: No module named 'datapipe.masks.train'; 'datapipe.masks' is not a package

zsyOAOA commented 3 months ago

solvers.py在这个project里面并没有使用，请忽略它就好！

yunjiefeng123 commented 3 months ago

尊敬的作者，您能单独发给我或者上传一份就相关工作用到的代码么？因为我是个小白，但是我的论文想做相关方面的东西，因为我看到别的一些.py文件也有类似的错误，比如models文件下边swin_transformer.py文件的第540行代码model = SwinIR(upscale=2, img_size=(height, width), 会报错NameError: name 'SwinIR' is not defined等错误，别的文件也有，所以能和您要一份完全用于您发表的那篇论文的代码么？如果有困扰的话就算了，谢谢！

zsyOAOA commented 3 months ago

这就是论文相对应的代码，按照readme提供的training和inference流程能够完全复现论文里的结果。之所以会出现你说的那些import error，是因为其中部分代码是从另外一个project里面copy过来的，没有完全清理干净。但是，在这个project的training和inference过程中，这些代码是完全使用不到的，你没有必要去关注和纠结这些冗余的部分。

如果你想快速熟悉我们的工作，就根据readme提供的指南去训练和测试就行，也只需要关注这些训练和测试的代码！

yunjiefeng123 commented 3 months ago

不好意思，又来打扰您，为什么我用了快速测试的python inference_resshift.py -i [image folder/image path] -o [result folder] --task bicsr --scale 4 这个命令，得到的结果仅是将图片放大了，有的人像还是模糊的，和您的结果不一样呢

zsyOAOA commented 3 months ago

不好意思，又来打扰您，为什么我用了快速测试的python inference_resshift.py -i [image folder/image path] -o [result folder] --task bicsr --scale 4 这个命令，得到的结果仅是将图片放大了，有的人像还是模糊的，和您的结果不一样呢

bicsr针对的是bicubic degradation, 对应于matlab下面的resize函数，如果degradation不match的话，效果就会很差。