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…