From 53742d024574d908c073dea15955487e5298198d Mon Sep 17 00:00:00 2001 From: Anastasiia Kutakh Date: Fri, 20 Aug 2021 21:45:27 -0400 Subject: [PATCH] . --- CDTools/models/base.py | 12 +++++++++++- CDTools/models/polarized_fancy_ptycho.py | 8 ++++++++ 2 files changed, 19 insertions(+), 1 deletion(-) diff --git a/CDTools/models/base.py b/CDTools/models/base.py index b200b3a..48a33d4 100644 --- a/CDTools/models/base.py +++ b/CDTools/models/base.py @@ -451,6 +451,16 @@ class CDIModel(t.nn.Module): """ + print('base models inspect: checking the object') + a = self.obj.detach() + def saveobj(a, filename): + a = np.abs(a) + plt.imshow(a) + plt.savefig('filename') + f = ['base_a.png', 'base_b.png', 'base_c.png', 'base_d.png'] + comp = [a[i, j, :, :] for i, j in zip([0, 0, 1, 1], [0, 1, 0, 1])] + for i in range(4): + saveobj(comp[i], f[i]) first_update = False if update and hasattr(self, 'figs') and self.figs: figs = self.figs @@ -585,7 +595,7 @@ class CDIModel(t.nn.Module): if logarithmic: sim_data =np.log(sim_data)/np.log(10) meas_data = np.log(meas_data)/np.log(10) - + if not updating: axes[0].set_title('Simulated') axes[1].set_title('Measured') diff --git a/CDTools/models/polarized_fancy_ptycho.py b/CDTools/models/polarized_fancy_ptycho.py index e81545e..93ae931 100644 --- a/CDTools/models/polarized_fancy_ptycho.py +++ b/CDTools/models/polarized_fancy_ptycho.py @@ -92,6 +92,13 @@ class PolarizedFancyPtycho(FancyPtycho): obj = t.stack((obj, obj), dim=-4) print('object', type(obj), obj.shape) model.obj.data = obj + print('polarized fancy ptycho from datset obj') + a = obj.detach() + plt.imshow(np.real(a[0, 0, :, :])) + plt.show() + plt.imshow(np.real(a[0, 1, :, :])) + plt.show() + # tensor vs tensor.data return model @@ -130,6 +137,7 @@ class PolarizedFancyPtycho(FancyPtycho): shift_probe=True, multiple_modes=True, polarized=True) analyzed_exit_waves = polarization.apply_linear_polarizer(exit_waves, analyzer) + # print('POLARIZED FANCY PTYCHO INTERACTION OBJ') return analyzed_exit_waves