-
Notifications
You must be signed in to change notification settings - Fork 411
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
ObjectDetectionTask: support for non-RGB imagery (SAR, MSI, HSI) #2602
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Copilot reviewed 3 out of 4 changed files in this pull request and generated no comments.
Files not reviewed (1)
- requirements/min-reqs.old: Language not supported
Comments suppressed due to low confidence (2)
torchgeo/trainers/detection.py:210
- The new behavior introduced by adapt_input_conv should be covered by tests to ensure it works as expected.
weight = adapt_input_conv(in_channels, self.model.backbone.body.conv1.weight) # type: ignore[no-untyped-call]
tests/trainers/test_detection.py:30
- Verify that the removal of the 'ObjectDetectionTestModel' and related monkey patches does not leave any gaps in test coverage.
class ObjectDetectionTestModel(Module):
5922fd1
to
7b9f02f
Compare
7b9f02f
to
be7f1f1
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Minor suggestion but otherwise lgtm
Co-authored-by: Isaac Corley <22203655+isaaccorley@users.noreply.github.com>
We don't yet have a non-RGB object detection dataset to test with, but I'm pretty sure this works. Needed to bump some deps and throw away some tests to get it to work.
Closes #1156