pixelshufflle实现超分辨率重建

最新推荐文章于 2024-09-12 19:01:23 发布

羊城迷鹿Ycml

最新推荐文章于 2024-09-12 19:01:23 发布

阅读量546

点赞数

文章标签： opencv 计算机视觉 pytorch

本文链接：https://blog.csdn.net/weixin_42432844/article/details/107009628

版权

本文介绍了在超分辨率重建中应用PixelShuffle算法的思路和过程，包括将图像转换为张量、增加通道维度以及使用PixelShuffle进行上采样。博主在实践中遇到了PIL版本问题和张量操作的挑战，如unsqueeze()和squeeze()的使用，并提供了相关参考资料链接以供深入学习。

摘要由CSDN通过智能技术生成

pixelshuffle定义：前人栽树：https://www.pianshen.com/article/33311325633/
算法思路：图像转张量（NxCxWxH）–增加C的维度–调用pixelshuffle
博主最近在做图像超分辨率重建，复现了SRCNN,RDN一些卷积之后concat再进行上采样的算法，今天做了一下简单的pixelshuffle。

import torch
from torch import nn
import cv2
import numpy as np
from torchvision import transforms

img1 = cv2.imread('./107.png',cv2.IMREAD_COLOR)

img1