resize by large image side

scripts/resize.py

@@ -9,29 +9,31 @@ def resize(images_dir, args, photos):
 
     log.ODM_DEBUG('Preparing images - %s' % images_dir)
 
+    # preconditions
     if len(photos) < 1:
         log.ODM_ERROR('Not enough photos in photos to resize')
         return False
 
-    # create directory for resized images
-    images_resize_dir = dataset.join_paths(images_dir, 'images_resize')
-    system.mkdir_p(images_resize_dir)
-
-    # define the new size
-    new_size = (args['resize_to'], args['resize_to'])
+    # create working directory
+    working_dir = dataset.join_paths(images_dir, 'images_resize')
+    system.mkdir_p(working_dir)
 
+    # loop over photos
     for photo in photos:
         try:
             # open and resize image with opencv
             img = cv2.imread(photo.path_file)
-            img_r = cv2.resize(img, new_size)
+            # compute new size
+            max_side = max(photo.width, photo.height)
+            ratio = float(args['resize_to']) / float(max_side)
+            img_r = cv2.resize(img, None, fx=ratio, fy=ratio)
             # write image with opencv
-            new_path_file = dataset.join_paths(images_resize_dir, photo.file_name)
+            new_path_file = dataset.join_paths(working_dir, photo.file_name)
             cv2.imwrite(new_path_file, img_r)
-            # read metadata
+            # read metadata with pyexiv2
             old_meta = pyexiv2.ImageMetadata(photo.path_file)
-            old_meta.read()
             new_meta = pyexiv2.ImageMetadata(new_path_file)
+            old_meta.read()
             new_meta.read()
             # copy metadata
             old_meta.copy(new_meta)
@@ -39,6 +41,9 @@ def resize(images_dir, args, photos):
             # log message
             log.ODM_DEBUG('Resized image %s | dimensions: %s' % \
                 (photo.file_name, img_r.shape))
+
+            # TODO(edgar): update photos array
+
         except cv2.error as e:
             # something went wrong with this image
             log.ODM_ERROR('Could not resize image %s' % photo.file_name)