在使用real-esrgan-x4plus模型对图片进行超分处理时,如何根据待处理的图片选择合适的模型呢?有哪些需要注意的呢?

弹道了无痕 2024-11-15 09:58:58

在使用real-esrgan-x4plus模型对图片进行超分处理时,如何根据待处理的图片选择合适的模型呢?有哪些需要注意的呢?

...全文
782 1 打赏 收藏 转发到动态 举报
AI 作业
写回复
用AI写文章
1 条回复
切换为时间正序
请发表友善的回复…
发表回复
weixin_38498942 2024-11-21
  • 打赏
  • 举报
回复

准备4个模型,输入尺寸分别为128x128、256x256、384x384和512x512,对不同输入尺寸的图片,需要根据尺寸合理地选择模型:

img

 

   def _select_model_size(self, height, width):
        if   width <= 128 and  height <= 128:
            return False, (128, 128)
        elif width <= 256 and  height <= 256:
            return False, (256, 256)
        elif width <= 384 and  height <= 384:
            return False, (384, 384)
        elif width <= 512 and  height <= 512:
            return False, (512, 512)
        elif width <= 512 or  height <= 512:
            return True,  (512, 512)
        return False, (-1, -1)

函数_select_model_size的返回值中包括两个值:布尔型的表示是否需要切分,二元组型的表示所选大模型。当需要对原图进行切分时,要优先按512x512进行切分,靠边的部分再选择合适尺寸的模型。
另外,为了避免对切分后超分处理的结果进行拼接时,不会出现如下图所示的竖纹,相邻两块切开需要至少16像素的重叠:

img

2,851

社区成员

发帖
与我相关
我的任务
社区描述
本论坛以AI、WoS 、XR、IoT、Auto、生成式AI等核心板块组成,为开发者提供便捷及高效的学习和交流平台。 高通开发者专区主页:https://qualcomm.csdn.net/
人工智能物联网机器学习 技术论坛(原bbs) 北京·东城区
社区管理员
  • csdnsqst0050
  • chipseeker
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告
暂无公告

试试用AI创作助手写篇文章吧