Skip to content

ObjectDetectionTask support for MSI #1156

Closed
@adamjstewart

Description

@adamjstewart

Summary

The ObjectDetectionTask has an in_channels parameter but it isn't actually used for anything. At the moment, it seems that the trainer only supports RGB imagery. We should fix this.

Rationale

We specialize in MSI, how can we not support MSI.

Implementation

We're currently using torchvision backbones, which makes things more challenging. How hard would it be to switch to timm backbones?

Alternatives

Alternatively, we'll have to override the first channel of the torchvision backbone.

Additional information

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    good first issueA good issue for a new contributor to work ontrainersPyTorch Lightning trainers

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions