From b9af0d283c78cd79582c01f87074061c6be29247 Mon Sep 17 00:00:00 2001 From: Jeremy Hummel Date: Sat, 22 May 2021 23:33:57 -0700 Subject: [PATCH] Update >256px previews --- models/Model_SAEHD/Model.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 8e91f0d..79ffa30 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -887,6 +887,23 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... st.append ( np.concatenate ( ar, axis=1) ) result += [ ('SAEHD pred', np.concatenate (st, axis=0 )), ] + wt = [] + for i in range(n_samples): + ar = SW[i], SS[i] + wt.append ( np.concatenate ( ar, axis=1) ) + result += [ ('SAEHD warped src-src', np.concatenate (wt, axis=0 )), ] + + wt = [] + for i in range(n_samples): + ar = DW[i], DD[i] + wt.append ( np.concatenate ( ar, axis=1) ) + result += [ ('SAEHD warped dst-dst', np.concatenate (wt, axis=0 )), ] + + wt = [] + for i in range(n_samples): + ar = DW[i], SD[i] + wt.append ( np.concatenate ( ar, axis=1) ) + result += [ ('SAEHD warped pred', np.concatenate (wt, axis=0 )), ] st_m = [] for i in range(n_samples):