From 84af9dc1ac8207b8c595b1d4ff19e49d7fc25b44 Mon Sep 17 00:00:00 2001 From: vpustova <vassili.pustovalov@irit.fr> Date: Fri, 25 Apr 2025 11:25:45 +0000 Subject: [PATCH] Upload New File --- DRPCA_Net/Test.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 DRPCA_Net/Test.py diff --git a/DRPCA_Net/Test.py b/DRPCA_Net/Test.py new file mode 100644 index 0000000..fba50af --- /dev/null +++ b/DRPCA_Net/Test.py @@ -0,0 +1,30 @@ +# -*- coding: utf-8 -*- +""" +Created on Wed Apr 23 17:44:09 2025 + +@author: vpustova +""" + +from admm_DRPCANet_v8 import RPCAnet +import torch +import numpy as np +from scipy.io import savemat +from scipy.io import loadmat + + + +device = 'cpu' +model=RPCAnet(kernel = [(9,1)]*20, admm_iterations=20) +state_dict=torch.load('DRPCAnet_100th_over_100epochs.pkl',map_location=device) +model.load_state_dict(state_dict) +model.eval() + +data=loadmat('D325.mat')['patch_180'] +data=data/np.max(np.abs(data)) +data = torch.tensor(data).to(device) +outputs_Lv,outputs_Sv=model(data) # Forward + +pred_test={'predS':outputs_Sv.cpu().detach().numpy(), 'predL':outputs_Lv.cpu().detach().numpy() } +savemat('D325processed.mat',pred_test) + + -- GitLab