pytorch与matlab,迁移pytorch工程至matlab

本文介绍了如何将基于PyTorch训练的单图像超分辨神经网络模型权重转换为MATLAB文件,并在MATLAB环境中复现网络的test过程。详细步骤包括模型加载、权重提取、权重转存以及MATLAB中PixelShuffle层的实现。
摘要由CSDN通过智能技术生成

好久没写博客了,险些以为自己找不到密码了。

最近抽空参与了个小项目,很惭愧,只做了三件小事

1. 基于PyTorch训练了一系列单图像超分辨神经网络

基于PyTorch训练了一系列单图像超分辨神经网络,超分辨系数从2-10。

该部分的实现参考了pytorch官方repo中的SR例程,训练程序包含于`./train`文件夹。该项目

基于高效子像素卷积层[1]进行空间分辨率提升操作,训练速度极快。

[1] ["Shi W, Caballero J, Huszar F, et al. Real-Time Single Image and

Video Super-Resolution Using an Efficient Sub-Pixel Convolutional

Neural Network[J]. 2016:1874-1883.](https://arxiv.org/abs/1609.05158)

2. 把训练好的模型权值转存为MATLAB文件。

简单粗暴,异常直接,只要把对应卷积层的权值全部提取出来就可以了。

提取的时候注意一点,要把pytorch中的Variable格式转换为Tensor,再转换为CPU模式,最终转换为numpy数组。

这一系列过程合并起来就是:

Var.data.cpu().numpy()

具体实现如下:

1 from __future__ importprint_function2

3 importtorch4 importnumpy as np5 importscipy.io as sio6

7 for i in [2, 3, 4, 5, 6, 7, 8, 9, 10]

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值