Update keypoint_dataloader.py

YunjiKim · Oct 29, 2019 · 738629d · 738629d
1 parent 8c0c7b6
commit 738629d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/data/keypoint_dataloader.py b/data/keypoint_dataloader.py
@@ -62,6 +62,7 @@ def _get_image_at(self, idx):
 
         # load images
         image = Image.open(osp.join(self._data_dir, img_path, '{:06d}'.format(1) + '.jpg'))
+        w,h = image.size
         crop_size, ratio = data_utils.center_crop(image, IMAGE_SIZE)
 
         image_seq = []