Skip to content

Commit

Permalink
improve MSE contrast in visualization
Browse files Browse the repository at this point in the history
  • Loading branch information
bunnie committed Feb 5, 2024
1 parent a1c8f00 commit 6b80287
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions overview.py
Original file line number Diff line number Diff line change
Expand Up @@ -473,6 +473,7 @@ def on_mse_visualize(self):
layers = list(self.layer_mse_norm_dict.keys())
mse_list = list(self.layer_mse_norm_dict.values())
mse_np = np.array(mse_list, dtype=float)
mse_np = 10 ** mse_np # undo the log10 weighting so the problems are more obvious in visualization
normalized_mse = cv2.normalize(mse_np, None, 0, 1, norm_type=cv2.NORM_MINMAX)
self.layer_mse_norm_dict = {key: value for key, value in zip(layers, normalized_mse)}
self.status_mse_visualize_button.setText('Remove MSE Overlay')
Expand Down

0 comments on commit 6b80287

Please sign in to comment.