From fcb0031a0440027ab9875e9b4b46546edeb8aa08 Mon Sep 17 00:00:00 2001 From: Pritika Date: Sun, 29 Mar 2026 19:55:03 +0530 Subject: [PATCH 1/2] Validate resize_interpolation in get_transform --- src/deepforest/model.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/src/deepforest/model.py b/src/deepforest/model.py index e667a03e1..e653bd16e 100644 --- a/src/deepforest/model.py +++ b/src/deepforest/model.py @@ -265,11 +265,15 @@ def get_transform(self, augmentations): resize_dims = self.config["cropmodel"].get("resize", [224, 224]) interp_name = self.config["cropmodel"].get("resize_interpolation", "bilinear") - interp = ( - transforms.InterpolationMode.NEAREST - if interp_name == "nearest" - else transforms.InterpolationMode.BILINEAR - ) + if interp_name == "nearest": + interp = transforms.InterpolationMode.NEAREST + elif interp_name == "bilinear": + interp = transforms.InterpolationMode.BILINEAR + else: + raise ValueError( + f"Invalid resize_interpolation '{interp_name}'. " + "Supported values are ['nearest', 'bilinear']." + ) data_transforms.append(transforms.Resize(resize_dims, interpolation=interp)) # Apply augmentations if specified From 73cfdd246a29b536f051b603daf005aa65dd1509 Mon Sep 17 00:00:00 2001 From: Pritika Date: Sun, 29 Mar 2026 21:45:17 +0530 Subject: [PATCH 2/2] Fix formatting issues (ruff) --- src/deepforest/model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/deepforest/model.py b/src/deepforest/model.py index e653bd16e..9a21dcf7d 100644 --- a/src/deepforest/model.py +++ b/src/deepforest/model.py @@ -285,7 +285,9 @@ def get_transform(self, augmentations): return transforms.Compose(data_transforms) - def expand_bbox_to_square(self, bbox, image_width, image_height): + def expand_bbox_to_square( + self, bbox: list[float], image_width: int, image_height: int + ) -> list[float]: """Expand a bounding box to a square by extending the shorter side. Parameters: