Lornatang / reality-sr

Real world super resolution library
Apache License 2.0
1 stars 0 forks source link

How can I get the checkpoint? #2

Open Explore008 opened 6 months ago

Explore008 commented 6 months ago

1709027012523

Explore008 commented 6 months ago

1709027012523

HI! I have solved this problem,but a new issue was produced. File "D:\project\reality-sr-main\demo\inference_images.py", line 40, in main inferencer = SuperResolutionInferencer(config_dict) File "D:\project\reality-sr-main\reality_sr\apis\super_resolution.py", line 41, in init self.model = SuperResolutionBackend(self.weights_path, self.device) File "D:\project\reality-sr-main\reality_sr\engine\backend.py", line 30, in init model = load_checkpoint(weights_path, map_location=device) File "D:\project\reality-sr-main\reality_sr\utils\checkpoint.py", line 94, in load_checkpoint checkpoint = torch.load(weights_path, map_location=map_location) File "D:\anaconda\envs\realsr\lib\site-packages\torch\serialization.py", line 789, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "D:\anaconda\envs\realsr\lib\site-packages\torch\serialization.py", line 1131, in _load result = unpickler.load() File "D:\anaconda\envs\realsr\lib\site-packages\torch\serialization.py", line 1124, in find_class return super().find_class(mod_name, name) ModuleNotFoundError: No module named 'real_esrgan' I do not know the means of real_esrgan module... Thank you for your reply!

Lornatang commented 6 months ago

Sorry, this project weights will not be released yet. You can try to use https://github.com/Lornatang/Real_ESRGAN-PyTorch

Explore008 commented 6 months ago

对不起,这个项目权重还没有公布。您可以尝试使用 https://github.com/Lornatang/Real_ESRGAN-PyTorch

Hi!could you provide a guide to training the model? I want to use my own datasets to get a checkpoints. The yaml for training is the inference.yaml? Thank you!!!