Shortcuts

Backbones and HeadsΒΆ

Backbones are the pre trained models that can be used with a task. The backbones or heads that are available can be found by using the available_backbones and available_heads methods.

To get the available backbones for a task like ImageClassifier, run:

from flash.image import ImageClassifier

# get the backbones available for ImageClassifier
backbones = ImageClassifier.available_backbones()

# print the backbones
print(backbones)

To get the available heads for a task like SemanticSegmentation, run:

from flash.image import SemanticSegmentation

# get the heads available for SemanticSegmentation
heads = SemanticSegmentation.available_heads()

# print the heads
print(heads)
Read the Docs v: latest
Versions
latest
stable
0.5.0
0.4.0
0.3.2
0.3.1
0.3.0
0.2.3
0.2.2
0.2.1
0.2.0
0.1.0post1
Downloads
pdf
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.