Which checkpoint file is the best?

fudan-zvg / SETR

[CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers

MIT License

1.05k stars 149 forks source link

Which checkpoint file is the best? #42

Closed Kittywyk closed 2 years ago

Kittywyk commented 3 years ago

Hi, thank you for your awesome work.

I trained SETR-MLA model on my own dataset, and there are many checkpoints files of different iterations. But how can I know which is the best one to test data?

Thanks in advance.

sixiaozheng commented 2 years ago

You can choose a model with a larger mIoU to test your own dataset according to the mIoU results of each model in the README. However, it is recommended that you train the model on your own dataset to obtain better test set results with high probability.