from flash.image import SemanticSegmentation
# get the heads available for SemanticSegmentation
heads = SemanticSegmentation.available_heads()
# print the heads
print(heads)
# [ ] # EMPTY!
from flash.image import SemanticSegmentation
# get the backbones available for SemanticSegmentation
backbones = SemanticSegmentation.available_backbones()
# print the backbones
print(backbones)
# [ ] # EMPTY!
🐛 Bug
flash.image.SemanticSegmentation class has no available heads or backbones.
To Reproduce
Code sample
From https://lightning-flash.readthedocs.io/en/stable/general/backbones.html#backbones-heads
From https://lightning-flash.readthedocs.io/en/stable/reference/semantic_segmentation.html
From https://github.com/Lightning-Universe/lightning-flash/blob/master/README.md
Expected behavior
SemanticSegmentation.available_heads() should return a list of heads. SemanticSegmentation.available_backbones() should return a list of backbones.
Environment
Issue exists in lightning-flash==0.8.1.post0 also
Additional context
ImageClassifier.available_heads() works as expected. ImageClassifier.available_backbones() works as expected.