[Bug fix & New feature] support non square sized input

[developer0hye]

Description

Support non square sized input

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)
• This change requires a documentation update

How has this change been tested, please provide a testcase or example of how you tested the change?

For a test, it needs some modification for test. But I didn't include these parts. I will submit PR for training, export and validation code for non square sized rf-detr.

import io
import requests
import supervision as sv
from PIL import Image
from rfdetr import RFDETRBase

model = RFDETRBase(resolution=(616, 560))
CLASS_NAMES = model.class_names

url = "https://media.roboflow.com/notebooks/examples/dog-2.jpeg"

image = Image.open(io.BytesIO(requests.get(url).content))
detections = model.predict(image, threshold=0.5)

labels = [
    f"{CLASS_NAMES[class_id]} {confidence:.2f}"
    for class_id, confidence
    in zip(detections.class_id, detections.confidence)
]

annotated_image = image.copy()
annotated_image = sv.BoxAnnotator().annotate(annotated_image, detections)
annotated_image = sv.LabelAnnotator().annotate(annotated_image, detections, labels)

sv.plot_image(annotated_image)

It failed to detect the person at the center point, which I believe is due to the model being trained only on square-shaped images. Still, it worked! This update marks the first step toward supporting non-square shapes.

Any specific deployment considerations

For example, documentation changes, usability, usage/costs, secrets, etc.

Docs

• Docs updated? What were the changes:
  Future works

[imerad]

This could be a very useful PR but I only see one commit which doesn't seem to include the changes on the screenshots. Did you forget to commit and push them ?

[developer0hye]

@imerad I think so. I didn't include other commits because I should implement the code for training to fully support non square inputs. But I think this commit will be the important point for supporting non square input.