Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

When will SegFormer and MiT be available? #1994

Closed
edge7 opened this issue Nov 27, 2024 · 4 comments
Closed

When will SegFormer and MiT be available? #1994

edge7 opened this issue Nov 27, 2024 · 4 comments

Comments

@edge7
Copy link

edge7 commented Nov 27, 2024

Hi,
I am trying to understand the situation a bit, I am fairly sure that one month ago I read a tutorial on the keras website showing how to use MiT backbone in a SegFormer for image segmentation.
Now am not able to find it anymore, probably because of Keras-Cv/Nlp --> Hub mapping.

I am not hugely expert in transformers for Computer Vision tasks, as I usually use CNN, but I wanted to try something else now.

At the minute there is nothing officially released here?

I just see MiT and SegFormer in the master branch but they have not been released.
Thanks for the clarificaitons

@edge7
Copy link
Author

edge7 commented Nov 27, 2024

I've installed the master branch and this works:

image_encoder = keras_hub.models.MiTBackbone.from_preset("mit_b1_ade20k_512", image_shape=(224, 224, 3))
backbone = keras_hub.models.SegFormerBackbone(image_encoder=image_encoder, projection_filters=256)
segformer = keras_hub.models.SegFormerImageSegmenter(backbone=backbone, num_classes=3)

@mattdangerw
Copy link
Member

@edge7 thanks for filing! Yes, some of the models from KerasCV are currently mid port and have not yet made it into a stable release of KerasHub. But MiT and SegFormer are both definitely things we are adding. We won't cut a release this week with the US holiday, but we should over the next week or two.

I'll check with the folks working on those models to see what remains (if anything), before they are ready for prime time.

@mattdangerw mattdangerw changed the title Help understanding keras-hub When will SegFormer and MiT be available? Nov 27, 2024
@edge7
Copy link
Author

edge7 commented Nov 27, 2024

Ok.
In the meantime, do you think that directly use the master branch is not particularly risky? At least to start testing it out

@mattdangerw
Copy link
Member

Ok. In the meantime, do you think that directly use the master branch is not particularly risky? At least to start testing it out

Not at all! Testing out via the master branch encouraged. pip install git+/~https://github.com/keras-team/keras-hub will build directly from master or pip install keras-hub-nightly for a nightly built package.

There might always be some changes before release.

@edge7 edge7 closed this as completed Dec 5, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants