Downloads · 30 days
0
Pranilllllll/segformer-b0-kaggle-ktm-dataset-random-fold0
segformer-b0-kaggle-ktm-dataset-random-fold0 is a machine learning model from Pranilllllll. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
Kathmandu Valley land-use segmentation, 6 classes, ignoreindex=255.
Downloads · 30 days
0
Access
Public
Updated Aug 20, 2026
Repo size
2.9 GB
Likes
0
Public
Click a slice to open those files.
.pt74.7 MB · 96%
From the Hugging Face model README
Kathmandu Valley land-use segmentation, 6 classes, ignore_index=255.
| field | value |
|---|---|
| architecture | SegformerForSemanticSegmentation, nvidia/segformer-b0-finetuned-ade-512-512 |
| split mode | random |
| fold | 0 of 3 |
| seed | 42 |
| input | 512x512, ImageNet norm, effective GSD 0.586 m/px |
| classes | Residential, Road, River, Forest, UnusedLand, Agricultural |
| lr (head/encoder) | 6e-05 / 6e-06 |
| regularization | wd 0.01, drop_path 0.1, smooth 0.05, EMA True |
| best epoch | 134 |
| val mIoU | 0.4948 |
| val mF1 | 0.6401 |
| val OA | 0.7863 |
| val kappa | 0.6707 |
| class | IoU | F1 |
|---|---|---|
| Residential | 0.7908 | 0.8832 |
| Road | 0.2984 | 0.4596 |
| River | 0.2422 | 0.3900 |
| Forest | 0.6428 | 0.7826 |
| UnusedLand | 0.4471 | 0.6180 |
| Agricultural | 0.5472 | 0.7074 |
Checkpoint best.pt holds model_state (EMA weights when EMA is on), the run cfg, the full HF model_config, and metrics.