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