segformer-b0-finetuned-segments-satellite-terrain

This model is a fine-tuned version of nvidia/mit-b0 on the ZeeeWP/terrain_map dataset. It achieves the following results on the evaluation set:

• Loss: 1.5487
• Mean Iou: 0.2261
• Mean Accuracy: 0.3435
• Overall Accuracy: 0.5155
• Accuracy Unlabeled: nan
• Accuracy Sand: 0.4844
• Accuracy Cliff: 0.7122
• Accuracy Bedrock flat: 0.6745
• Accuracy Bedrock lowhill: 0.0645
• Accuracy Bedrock highhill: 0.0
• Accuracy Gravel low hill: 0.4691
• Accuracy Gravel high hill: 0.0
• Iou Unlabeled: nan
• Iou Sand: 0.4487
• Iou Cliff: 0.4859
• Iou Bedrock flat: 0.3646
• Iou Bedrock lowhill: 0.0584
• Iou Bedrock highhill: 0.0
• Iou Gravel low hill: 0.2249
• Iou Gravel high hill: 0.0

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 6e-05
• train_batch_size: 2
• eval_batch_size: 2
• seed: 42
• optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• lr_scheduler_type: linear
• num_epochs: 10

Training results

Training Loss	Epoch	Step	Validation Loss	Mean Iou	Mean Accuracy	Overall Accuracy	Accuracy Unlabeled	Accuracy Sand	Accuracy Cliff	Accuracy Bedrock flat	Accuracy Bedrock lowhill	Accuracy Bedrock highhill	Accuracy Gravel low hill	Accuracy Gravel high hill	Iou Unlabeled	Iou Sand	Iou Cliff	Iou Bedrock flat	Iou Bedrock lowhill	Iou Bedrock highhill	Iou Gravel low hill	Iou Gravel high hill
1.7462	2.5	20	1.9462	0.2019	0.3088	0.4750	nan	0.3936	0.6995	0.6335	0.0725	0.0	0.3624	0.0	nan	0.3812	0.4667	0.3481	0.0591	0.0	0.1582	0.0
1.6774	5.0	40	1.7093	0.2078	0.3221	0.4960	nan	0.4227	0.7176	0.7059	0.0361	0.0	0.3723	0.0	nan	0.4017	0.4888	0.3484	0.0329	0.0	0.1826	0.0
1.3793	7.5	60	1.5717	0.2235	0.3421	0.5081	nan	0.4908	0.6816	0.6798	0.0656	0.0	0.4769	0.0	nan	0.4458	0.4760	0.3635	0.0583	0.0	0.2210	0.0
1.7613	10.0	80	1.5487	0.2261	0.3435	0.5155	nan	0.4844	0.7122	0.6745	0.0645	0.0	0.4691	0.0	nan	0.4487	0.4859	0.3646	0.0584	0.0	0.2249	0.0

Framework versions

• Transformers 4.44.1
• Pytorch 2.4.1
• Datasets 2.14.6
• Tokenizers 0.19.1