SegFormer Weight Unavailable

[StephenYangjz]

Greetings,

I think the original repo's gdrive link for the checkpoint segformer.b5.1024x1024.city.160k.pth has been taken down. I tried to look around but havent been able to find any other places to download it. Would it be possible for anyone to kindly help me with getting the weights? Thank you so much : ) @cyrilzakka @Chrisding

[wyf0824]

You can go to Huggingface and Model Scope to download it. :)

[YangWangforever]

Thank you so much!@wyf0824

[YangWangforever]

nvidia/mit-b0 · Hugging Face here it is! YOU ARE WELCOME!

------------------ 原始邮件 ------------------ 发件人: "NVlabs/SegFormer" @.>; 发送时间: 2025年10月18日(星期六) 上午6:47 @.>; @.@.>; 主题: Re: [NVlabs/SegFormer] SegFormer Weight Unavailable (Issue #165)

bahadirakman left a comment (NVlabs/SegFormer#165)

Thank you so much

could you please send me a link of the weights, i can not find it

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>

[YangWangforever]

https://huggingface.co/nvidia/mit-b0

[YangWangforever]

I am back here again, it turns out that mit_b5 on hugging face is a bit of different from the original, the key in model.named_parameter() is quite different,and the structure is a bit of differrent. So the hugging face version cannot be used to train this model. Can anyone provide an accessible way to download the original mit_b5.pth?

[Shelfcol]

You can download segformer.b5.1024x1024.city.160k.pth using command（gdown 1e7DECAH0TRtPZM6hTqRGoboq1XPqSmuj）， from drivestudio waymo installation and process @YangWangforever

[hover03]

You can download segformer.b5.1024x1024.city.160k.pth using command（gdown 1e7DECAH0TRtPZM6hTqRGoboq1XPqSmuj）， from drivestudio waymo installation and process @YangWangforever

thanks a lot! But how to download other checkpoint like b0...? I have checked the website of waymo and it seems that they mention the backup id of b5 only.....

[MikeDegany]

You can download it from here: https://huggingface.co/nvidia/segformer-b5-finetuned-cityscapes-1024-1024/tree/main

[hover03]

thanks a lot for your response(^▽^)

[yulnn]

The weights on hugging face still have key mismatches. Can anyone provide or point to a download location for the properly trained mit_b5.pth file?

[CaMi1le]

You can download segformer.b5.1024x1024.city.160k.pth using command（gdown 1e7DECAH0TRtPZM6hTqRGoboq1XPqSmuj）， from drivestudio waymo installation and process @YangWangforever

Great solutions, it works.