diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index 9904111a02df849b78e90aab30b6b567ba4b1ec4..2b47bd63cde932cec021b9e37b1cdd269faed601 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -802,6 +802,8 @@ def analyze(file='/Users/tomrink/cld_opd_out.npy'): pred = pred[:, :, :, 0] print('Total num pixels: ', lbls.size) + pred = pred.flatten() + lbls = lbls.flatten() diff = pred - lbls mae = (np.sum(np.abs(diff))) / diff.size @@ -817,7 +819,7 @@ def analyze(file='/Users/tomrink/cld_opd_out.npy'): bin_edges.append(0.1) bin_ranges.append([0.5, 2.0]) - bin_edges.append([0.5]) + bin_edges.append(0.5) bin_ranges.append([2.0, 5.0]) bin_edges.append(2.0) @@ -857,7 +859,7 @@ def analyze(file='/Users/tomrink/cld_opd_out.npy'): bin_edges.append(160.0) - diff_by_value_bins = util.util.bin_data_by(diff.flatten(), lbls.flatten(), bin_ranges) + diff_by_value_bins = util.util.bin_data_by(diff, lbls, bin_ranges) values = [] for k in range(len(bin_ranges)):