Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
yunjiefeng123 opened this issue Jun 9, 2024 · 7 comments
Open

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

yunjiefeng123 opened this issue Jun 9, 2024 · 7 comments

Comments

@yunjiefeng123
Copy link

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

@zsyOAOA
Copy link
Owner

zsyOAOA commented Jun 9, 2024

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

@yunjiefeng123
Copy link
Author

就在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
Copy link
Owner

zsyOAOA commented Jun 9, 2024

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

@yunjiefeng123
Copy link
Author

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

@zsyOAOA
Copy link
Owner

zsyOAOA commented Jun 11, 2024

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

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

@yunjiefeng123
Copy link
Author

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

@zsyOAOA
Copy link
Owner

zsyOAOA commented Jun 25, 2024

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

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants