[功能] timm.models.adapt_input_conv: 除了 RGB 权重之外

作者: adamjstewart创建于 2025年2月25日更新于 2025年2月26日
标签enhancement

您的功能请求是否与问题有关?请描述。

TorchGeo 提供了在非 RGB 图像(例如 Sentinel-2, 13 通道)上预先训练的多个模型权重。在处理时间序列数据时,我们通常希望沿通道维度堆叠图像,以便最终得到 $$B \times TC \times H \times W$$ 的输入。然而,我们尚未找到一种简单的方法来调整我们的预训练权重以匹配。

内容来源: huggingface/pytorch-image-models