rstrudel / segmenter

[ICCV2021] Official PyTorch implementation of Segmenter: Transformer for Semantic Segmentation
MIT License
842 stars 174 forks source link

replicate cityscapes performance #41

Closed ekellbuch closed 2 years ago

ekellbuch commented 2 years ago

Hi,

Thanks for making the code open source. I was wondering if you could share the config file you used to train a seg-l/16 model with a vit-l/16 backbone to achieve 80.7% for the cityscapes dataset (second to last row in table 8). When I tried using the same configs as https://www.rocq.inria.fr/cluster-willow/rstrudel/segmenter/checkpoints/cityscapes/seg_large_mask/variant.yml but replacing the decoder the performance I got was around 75%. Thanks.

rstrudel commented 2 years ago

Hi @ekellbuch , We released Seg-L/16 checkpoint at the following directory: https://www.rocq.inria.fr/cluster-willow/rstrudel/segmenter/checkpoints/cityscapes/seg_large_linear/

I hope this can help you reproduce it :) Robin

rstrudel commented 2 years ago

I am closing the issue as it is stale.