Questions tagged [segformer]
2 questions
1
vote
4 answers
Loss is Nan for SegFormer vision transformer trained on BDD10k
I'm trying to implement a SegFormer pretrained with a mit-b0 model to perform semantic segmentation on images obtained from the bdd100k dataset. Specifically, semantic segmentation has masks for only a subset of the 100k images, being 10k with…

Jimenemex
- 3,104
- 3
- 24
- 56
0
votes
0 answers
How to load a SegFormer pytorch checkpointfile and visualize a prediction through it?
I have trained a segformer model using the code in this article (courtesy of Roboflow)
https://blog.roboflow.com/how-to-train-segformer-on-a-custom-dataset-with-pytorch-lightning/'
I end up with a checkpoint file, and now I want a seperat code to…

Jesper Andersen
- 1
- 2