Skip to content

Commit

Permalink
docs: update self_heatmap.py
Browse files Browse the repository at this point in the history
interations -> interactions
  • Loading branch information
eltociear authored Aug 29, 2024
1 parent 7deca81 commit 9f092ac
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions alfie/self_heatmap.py
Original file line number Diff line number Diff line change
Expand Up @@ -254,7 +254,7 @@ def compute_pixel_diffused_heat_map(self,
latent_pixel_id: The latent pixel id from which the heatmap will be diffused throughout the latent image
method: Currently only has `thresholding` where at each step to remove the misguiding/noisy pixels to
prevent them from focussing/enhancing wrong heatmaps, we use simple thresholding
n_iter: For how many interations to refine the heatmap as described above
n_iter: For how many interactions to refine the heatmap as described above
plot_itermediate: If the intermediate heatmaps need to plotted to show the evolutation
"""
Expand Down Expand Up @@ -288,7 +288,7 @@ def compute_diffused_heat_map(self,
`n_iter` number of iterations. Returns a SelfGlobalHeatMap collection of the attention diffused heatmaps for all pixels
method: Currently only has `thresholding` where at each step to remove the misguiding/noisy pixels to
prevent them from focussing/enhancing wrong heatmaps, we use simple thresholding
n_iter: For how many interations to refine the heatmap as described above
n_iter: For how many interactions to refine the heatmap as described above
"""
# convert the latent 2d image from height.width x height x weight to 1 x height.weight x height x weight
Expand Down Expand Up @@ -346,4 +346,4 @@ def __iter__(self):

def clear(self):
self.ids_to_heatmaps.clear()
self.ids_to_num_maps.clear()
self.ids_to_num_maps.clear()

0 comments on commit 9f092ac

Please sign in to comment.