Skip to content

Commit

Permalink
fix variable names, use native min method
Browse files Browse the repository at this point in the history
  • Loading branch information
pleary committed May 16, 2024
1 parent 5d5a3ce commit 1e07500
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions lib/inat_inferrer.py
Original file line number Diff line number Diff line change
Expand Up @@ -563,10 +563,10 @@ def prepare_image_for_inference(file_path):

eventual_size = 299
central_crop_factor = 0.875
resize_min_dimension = eventual_size/central_crop_factor
resize_min_dimension = eventual_size / central_crop_factor

height, width = image.shape[0], image.shape[1]
resize_ratio = np.array([height, width]).min() / resize_min_dimension
resize_ratio = min(height, width) / resize_min_dimension
new_height = math.ceil(height / resize_ratio)
new_width = math.ceil(width / resize_ratio)
# resize the image so we can take a central crop without needing to resample again
Expand All @@ -577,11 +577,11 @@ def prepare_image_for_inference(file_path):
preserve_aspect_ratio=True
)
# determine the upper-left corner that needs to be used to grab the square crop
upper_left = math.floor((new_height - eventual_size) / 2)
upper_right = math.floor((new_width - eventual_size) / 2)
offset_height = math.floor((new_height - eventual_size) / 2)
offset_width = math.floor((new_width - eventual_size) / 2)
# take a square crop out of the resized image
image = tf.image.crop_to_bounding_box(
image, upper_left, upper_right, eventual_size, eventual_size
image, offset_height, offset_width, eventual_size, eventual_size
)
return tf.expand_dims(image, 0)

Expand Down

0 comments on commit 1e07500

Please sign in to comment.