Skip to content

Segformer Support #382

Description

@HugeBob

Feature request

Would love for Optimum to add support for transformers.SegformerForSemanticSegmentation

https://huggingface.co/docs/transformers/model_doc/segformer#transformers.SegformerForSemanticSegmentation

As best I could tell, semantic segmentation is not something that Optimum currently supports for any models (https://huggingface.co/docs/optimum/main/en/pipelines) would love for this to be improved!

Motivation

I use HuggingFace's Segformer for an image segmentation model I have and would love to improve my inference speeds.

Your contribution

I don't know what a PR is so I kind of doubt it.

Metadata

Metadata

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions