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 27 Packages Projects Releases Wiki Activity
21e23ffca7
transformers/docs/source/en/main_classes/backbones.md
amyeroberts 7252e8d937
[Doc] Fix docbuilder - make BackboneMixin and BackboneConfigMixin importable from utils. (#29002) 
* Trigger doc build

* Test removing references

* Importable from utils

* Trigger another run on a new commit for testing
2024-02-14 10:29:22 +00:00

2.2 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.BackboneMixin] enables initializing a backbone from Transformers or timm and includes functions for returning the output features and indices.
  • [~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.BackboneMixin

BackboneConfigMixin

autodoc utils.BackboneConfigMixin

TimmBackbone

autodoc models.timm_backbone.TimmBackbone

TimmBackboneConfig

autodoc models.timm_backbone.TimmBackboneConfig