Merge pull request #565 from sheth2chintan/fix_demo_pad

update augmentation.py
AliaksandrSiarohin · May 4, 2023 · c6996b9 · c6996b9
2 parents 3d152de + 1242092
commit c6996b9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/augmentation.py b/augmentation.py
@@ -9,7 +9,7 @@
 import PIL
 
 from skimage.transform import resize, rotate
-from skimage.util import pad
+from numpy import pad
 import torchvision
 
 import warnings

diff --git a/logger.py b/logger.py
@@ -4,7 +4,7 @@
 import imageio
 
 import os
-from skimage.draw import circle
+from skimage.draw import disk
 
 import matplotlib.pyplot as plt
 import collections
@@ -111,7 +111,7 @@ def draw_image_with_kp(self, image, kp_array):
         kp_array = spatial_size * (kp_array + 1) / 2
         num_kp = kp_array.shape[0]
         for kp_ind, kp in enumerate(kp_array):
-            rr, cc = circle(kp[1], kp[0], self.kp_size, shape=image.shape[:2])
+            rr, cc = disk(kp[1], kp[0], self.kp_size, shape=image.shape[:2])
             image[rr, cc] = np.array(self.colormap(kp_ind / num_kp))[:3]
         return image