欢迎光临散文网 会员登陆 & 注册

pytorch-nmf实现

2022-12-02 21:20 作者:021usc  | 我要投稿

参考

https://yoyololicon.github.io/posts/2021/02/torchnmf-algorithm/

https://github.com/yoyololicon/pytorch-NMF

https://gitee.com/youryouth/mc/tree/master/nmf

NMF分解

这里是通过下载别人写好的库实现的。首先安装对应的库

然后通过下述代码,实现单层的矩阵分解

单层NMF实现的结果

上图左为原始图片,上图右为矩阵分解后补全的图片,可以看出分解后的图片在某些细节处理上并不是那么好。

多层NMF分解

深度半矩阵分解

将矩阵X进行深度分解,即X%20%3D%20W_1W_2H_2

三层矩阵分解效果图

批量NMF分解

批量NMF分解,秩为10

三通道批量NMF分解

三通道批量NMF分解,秩为10


pytorch-nmf实现的评论 (共 条)

分享到微博请遵守国家法律