參考

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

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

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

NMF分解

這里是通過下載別人寫好的庫實(shí)現(xiàn)的。首先安裝對(duì)應(yīng)的庫

然后通過下述代碼，實(shí)現(xiàn)單層的矩陣分解

上圖左為原始圖片，上圖右為矩陣分解后補(bǔ)全的圖片，可以看出分解后的圖片在某些細(xì)節(jié)處理上并不是那么好。

多層NMF分解

將矩陣進(jìn)行深度分解，即。

批量NMF分解

三通道批量NMF分解