resize_with_crop_or_pad gives better overall results

see inaturalist/iNaturalistMLWork#24 for some results
inaturalist · Mar 19, 2024 · b5ab344 · b5ab344
1 parent 5af2e58
commit b5ab344
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/lib/inat_inferrer.py b/lib/inat_inferrer.py
@@ -364,8 +364,9 @@ def prepare_image_for_inference(file_path):
             image = tf.io.read_file(file_path)
             image = tf.image.decode_jpeg(image, channels=3)
         image = tf.image.convert_image_dtype(image, tf.float32)
-        image = tf.image.central_crop(image, 0.875)
-        image = tf.image.resize(image, [299, 299], tf.image.ResizeMethod.NEAREST_NEIGHBOR)
+        image = tf.image.resize_with_crop_or_pad(
+            image, 299, 299
+        )
         return tf.expand_dims(image, 0)
 
     @staticmethod

diff --git a/lib/vision_testing.py b/lib/vision_testing.py
@@ -153,8 +153,9 @@ def prepare_image_for_inference(self, cache_path):
         image = tf.io.read_file(cache_path)
         image = tf.image.decode_jpeg(image, channels=3)
         image = tf.image.convert_image_dtype(image, tf.float32)
-        image = tf.image.central_crop(image, 0.875)
-        image = tf.image.resize(image, [299, 299], tf.image.ResizeMethod.NEAREST_NEIGHBOR)
+        image = tf.image.resize_with_crop_or_pad(
+            image, 299, 299
+        )
         return tf.expand_dims(image, 0)
 
     def assess_top_results(self, observation, top_results):