PyPRIS/ts.py at master · xiyuyi/PyPRIS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import matplotlib.pyplot as plt
from PyPRIS import *
f = list()
f.append("bgSCF0.8_mu1.0e+09")
f.append("bgSCF0.8_mu1.0e+10")
f.append("bgSCF0.8_mu1.0e+11")
f.append("bgSCF0.8_mu1.0e+12")
f.append("bgSCF0.8_mu8.0e-01")
f.append("bgSCF1.5_mu1.0e+09")
f.append("bgSCF1.5_mu1.0e+10")
f.append("bgSCF1.5_mu1.0e+11")
f.append("bgSCF1.5_mu1.0e+12")
f.append("bgSCF1.5_mu1.5e+00")
f.append("bgSCF1_mu1.0e+00")
f.append("bgSCF1_mu1.0e+09")
f.append("bgSCF1_mu1.0e+10")
f.append("bgSCF1_mu1.0e+11")
f.append("bgSCF1_mu1.0e+12")
f.append("bgSCF2_mu1.0e+09")
f.append("bgSCF2_mu1.0e+10")
f.append("bgSCF2_mu1.0e+11")
f.append("bgSCF2_mu1.0e+12")
f.append("bgSCF2_mu2.0e+00")

itN = 0
fitem = f[14]

PyPRIS_SensMx_name = "PyPRIS_"+fitem+"_pris0_SensingMx" # specify sensing matrix file name.
path = "G:/DH_localization/PyPRIS_tickets_set3/"+fitem+"/saved_objects"  # specifie datafile position
PyPRIS_name = "PyPRIS_"+fitem+"_pris0_"+str(1+itN)  # specify datafile name
print(path+"/"+PyPRIS_name)
try:
    linbreg = loadCSSolver(path, PyPRIS_name, PyPRIS_SensMx_name)
    linbreg.path_d = path
    linbreg.debug = True
    try:
        linbreg.debug_output(linbreg.it_count,'visualize')
    except:
        print("canpt output plots")
        pass
    v = linbreg.candidate_vis()
    vis = v[:,:,:]
    prj_ax0 = copy.deepcopy(np.mean(vis, axis=0))
    prj_ax1 = copy.deepcopy(np.mean(vis, axis=1))
    prj_ax2 = copy.deepcopy(np.mean(vis, axis=2).T)
    patch = np.zeros((vis.shape[0],vis.shape[0]))
    plt.figure(figsize=(3,3))
    cat1 = np.concatenate([prj_ax0, prj_ax2], axis = 1)
    cat2 = np.concatenate([prj_ax1, patch], axis = 1)
    cat = np.concatenate([cat1, cat2], axis = 0)
    plt.imshow(cat)
    plt.savefig(
        '{}/PyPRIS__{}_{}_{}_plots_it{}.png'.format( linbreg.path_d, 'Proj_vies', linbreg.PyPRIS_name, linbreg.PyPRIS_iter, linbreg.it_count),
        dpi=300, figsize=(100, 80))
except:
    pass