Skip to content
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

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

adamjstewart
Copy link
Collaborator

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

@adamjstewart adamjstewart added this to the 0.6.3 milestone Feb 21, 2025
@github-actions github-actions bot added testing Continuous integration testing trainers PyTorch Lightning trainers dependencies Packaging and dependencies labels Feb 21, 2025
@calebrob6 calebrob6 requested a review from Copilot February 21, 2025 21:37

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):
@github-actions github-actions bot added the models Models and pretrained weights label Feb 21, 2025
@adamjstewart adamjstewart changed the title ObjectDetectionTask: support for MSI, HSI, SAR ObjectDetectionTask: support for SAR, MSI, HSI (non-RGB imagery) Feb 23, 2025
@adamjstewart adamjstewart changed the title ObjectDetectionTask: support for SAR, MSI, HSI (non-RGB imagery) ObjectDetectionTask: support for non-RGB imagery (SAR, MSI, HSI) Feb 25, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
dependencies Packaging and dependencies models Models and pretrained weights testing Continuous integration testing trainers PyTorch Lightning trainers
Projects
None yet
Development

Successfully merging this pull request may close these issues.

ObjectDetectionTask support for MSI
1 participant