diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py index e5e0b6e62793173acedcb4c0b443103572c2d466..9a80f99c8ee4bf1d4af82745c875aecba9881675 100644 --- a/modules/deeplearning/srcnn_l1b_l2.py +++ b/modules/deeplearning/srcnn_l1b_l2.py @@ -780,8 +780,8 @@ def analyze(lbls, pred): print('Total num pixels: ', lbls.size) pred = pred.flatten() - pred = np.where(pred < 0.0, 0.0, pred) - pred = np.where(pred > 160.0, 160.0, pred) + pred = np.where(pred < 0.0, 0.5, pred) + pred = np.where(pred > 160.0, 158.0, pred) lbls = lbls.flatten() diff = pred - lbls @@ -842,8 +842,8 @@ def analyze(lbls, pred): # bin_edges.append(160.0) # ----------------------------------- - bin_ranges.append([0.01, 2.0]) - bin_edges.append(0.01) + bin_ranges.append([0.5, 2.0]) + bin_edges.append(0.5) bin_ranges.append([2.0, 5.0]) bin_edges.append(2.0)