search

NVlabs / SegFormer

Official PyTorch implementation of SegFormer
https://arxiv.org/abs/2105.15203
Other
2.46k stars 345 forks source link

Add convolution block before the entire model #65

Open Shravan-3501 opened 2 years ago

Shravan-3501 commented 2 years ago

How can I modify the model to add a convolution block before the first conv layer?

YScheung commented 2 years ago

Same Question. Is there a way that we could access the SegFormer model architecture ?