riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-05 13:50:13 +06:00
Code Issues Actions 26 Packages Projects Releases Wiki Activity
3f9f749325
transformers/docs/source/en/main_classes/backbones.md
Steven Liu abbffc4525
[docs] Backbone (#28739) 
* backbones

* fix path

* fix paths

* fix code snippet

* fix links
2024-02-01 09:16:16 -08:00

2.3 KiB
Raw Blame History

Backbone

A backbone is a model used for feature extraction for higher level computer vision tasks such as object detection and image classification. Transformers provides an [AutoBackbone] class for initializing a Transformers backbone from pretrained model weights, and two utility classes:

  • [~utils.backbone_utils.BackboneMixin] enables initializing a backbone from Transformers or timm and includes functions for returning the output features and indices.
  • [~utils.backbone_utils.BackboneConfigMixin] sets the output features and indices of the backbone configuration.

timm models are loaded with the [TimmBackbone] and [TimmBackboneConfig] classes.

Backbones are supported for the following models:

AutoBackbone

autodoc AutoBackbone

BackboneMixin

autodoc utils.backbone_utils.BackboneMixin

BackboneConfigMixin

autodoc utils.backbone_utils.BackboneConfigMixin

TimmBackbone

autodoc models.timm_backbone.TimmBackbone

TimmBackboneConfig

autodoc models.timm_backbone.TimmBackboneConfig