Yang Fan

How to direct SR on images without preliminary degradation

1

Hello, I'm currently working with your SR processing tool and have a question regarding its usage with custom images. According to the README.md, the process involves degrading images before applying...

使用tiled对大图像执行SR时失败了

1

非常感谢您的作品。 在使用参数 `--tiled --tile_size 512 --tile_stride 256`，并在执行前使用了 ``` export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:64 ``` 对1024px\*1024px的图像进行SR（V2）时，在24G显存和48G显存的机器上均可以成功执行。 但当对1536px\*1536px的图像进行SR（V2）时，在24G显存和48G显存的机器上均无法执行。 错误信息如下所示，该错误信息是在48G显存的机器上出现的，它尝试申请81GB内存，注定是失败了。 在24G显存的机器上执行时，它尝试申请9GB内存，但仍然失败了。 对同一张图像处理时，申请的内存大小相差甚远，这点感觉也很奇怪。 通过代码了解到仅在Stage2时用到了tiled选项，而在Stage1时是对整个图像进行处理。 请问是否有办法解决呢 十分感谢，期待您的回复！ ``` Traceback (most recent call last): File "/home/yangfan/work/DiffBIR/inference.py", line...