It turns out that nvidia/segformer-b4-finetuned-ade-512-512 on hugging face is a bit of different from the your code implement. <img width="1492" height="518" alt="Image" src="https://github.com/user-attachments/assets/f8a98228-3cca-4371-bdef-f48de46b46a6" />
It turns out that nvidia/segformer-b4-finetuned-ade-512-512 on hugging face is a bit of different from the your code implement.