Skip to content

Finalize EdgeCrafter backbones package implementation #928

Open

Description

@liopeer
  • adapt the existing feature pyramid to the public interface
  • enable support for multi-channel images following the other packages

Activity

  1. lorinczszabolcs commented on Aug 22, 2026

    @lorinczszabolcs
    Contributor

    Hi @liopeer, I addressed the second item of this issue: multi-channel image support, following the other packages.

    What I did:

    • EdgeCrafterPackage now accepts num_input_channels and builds the ECViT patch embed for that many channels. It no longer restricts the input to 3 channels.
    • Pretrained RGB weights are adapted to the requested channel count on load, through a load_state_dict pre-hook that reuses the shared helper the DINOv2, DINOv3, and TIMM packages use.

    Note on the stack: this change is stacked on #950 (the first item). GitHub does not let a fork pull request use another fork branch as its base. So this PR is opened against main and shows both commits until #950 merges. After #950 is merged I will rebase this branch onto main so it contains only the multi-channel commit.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions