sail-sg / poolformer

PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)
https://arxiv.org/abs/2111.11418
Apache License 2.0
1.29k stars 117 forks source link

PoolFormer for Segmentation task #56

Open catchyoo opened 9 months ago

catchyoo commented 9 months ago

Hello! When applying PoolFormer to a segmentation task, using the provided pth file shows that there are no weights for the hierarchical features used for dense prediction.

Since I use a pretrained model for classification tasks, I think this is an expected result.

Did you also train PoolFormer for segmentation task without weights for hierarchical features?

Thank you.