Simple SegFormer network class

[rose-jinyang]

Hello
How are you?
Thanks for contributing to this project.
It is difficult for us to use this project because it contains many other scripts.
Did u check https://github.com/lucidrains/segformer-pytorch which is a third-party implementation for SegFormer?
This project contains ONLY a simple segformer network class so it is easy to use.
But the number of params of MiT-B0 network by this implementation is 7M.
I know that the number of params of MiT-B0 is 3.6M in the paper.
Could u check https://github.com/lucidrains/segformer-pytorch shortly?
If it is difficult, could u make the SegFormer network class like the above implementation?
Thanks

[xieenze]

Hi,
Thanks for your interest. I have checked the repo you referred. It is easy to read, but I think it only contains the network part, which is not enough for one to train and evaluate on several datasets.

On the other hand, openmmlab's mmsegmentation is a very advanced toolbox for semantic segmentation, it also has detailed documents for a beginner to easily get started.

So I prefer to use mmsegmentation to implement the SegFormer, without mmsegmentation, SegFormer cannot easily support several datasets and different encoders.
You may first learn how to use mmsegmentation following the get_started.md. I believe it is easy to figure it out.

[rose-jinyang]

Thanks for your reply.
Could u check https://github.com/lucidrains/segformer-pytorch shortly if there is any issue for MiT B0?
If it is difficult, could u make the simple SegFormer network class like the above implementation?
Thanks

[xieenze]

Will consider checking it when I am free.

[rose-jinyang]

Could u make the simple SegFormer network class like the above implementation?

[xieenze]

Could u make the simple SegFormer network class like the above implementation?

Will consider to do it when I have time :)

[rose-jinyang]

Thanks