Rename

open-mmlab · May 31, 2021 · 7246772 · 7246772
1 parent 367a5ff
commit 7246772
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmedit/datasets/pipelines/generate_assistant.py b/mmedit/datasets/pipelines/generate_assistant.py
@@ -65,8 +65,8 @@ def _generate_one_heatmap(self, keypoint):
         x_range = np.arange(start=0, stop=w, dtype=int)
         y_range = np.arange(start=0, stop=h, dtype=int)
         grid_x, grid_y = np.meshgrid(x_range, y_range)
-        dist = (grid_x - keypoint[0])**2 + (grid_y - keypoint[1])**2
-        exponent = dist / 2.0 / self.sigma / self.sigma
+        dist2 = (grid_x - keypoint[0])**2 + (grid_y - keypoint[1])**2
+        exponent = dist2 / 2.0 / self.sigma / self.sigma
         heatmap = np.exp(-exponent)
         return heatmap