segformer-b0-finetuned-segments-ECHO-dev-05-v1

unreal-hug/segformer-b0-finetuned-segments-ECHO-dev-05-v1

segformer-b0-finetuned-segments-ECHO-dev-05-v1 is a image segmentation model from unreal-hug. Use it for the image segmentation task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as other.

should probably proofread and complete it, then remove this comment. --

Downloads · 30 days

8% of all-time downloads

All-time downloads

Public

Parameters

3.7M

15 MB on disk

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.safetensors14.9 MB · 99%

At a glance

Task: Image Segmentation
Library: transformers
License: other
Model type: segformer
Access: Public
Created: Dec 5, 2023
Updated: Dec 5, 2023
SHA: 89d7e121

Base models

nvidia/mit-b0

Model card

From the Hugging Face model README

segformer-b0-finetuned-segments-ECHO-dev-05-v1

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

Loss: 0.4592
Mean Iou: 0.3826
Mean Accuracy: 0.5892
Overall Accuracy: 0.5467
Accuracy Unlabeled: nan
Accuracy Lv: 0.7143
Accuracy Lr: 0.4323
Accuracy Ra: 0.7629
Accuracy La: 0.4472
Iou Unlabeled: 0.0
Iou Lv: 0.7065
Iou Lr: 0.4317
Iou Ra: 0.4223
Iou La: 0.3527

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: 50

Training results

Training Loss	Epoch	Step	Validation Loss	Mean Iou	Mean Accuracy	Overall Accuracy	Accuracy Unlabeled	Accuracy Lv	Accuracy Lr	Accuracy Ra	Accuracy La	Iou Lv	Iou Lr	Iou Ra	Iou La
1.1252	2.86	20	1.3259	0.1971	0.3379	0.4375	nan	0.0	0.6365	0.5291	0.1860	0.0	0.4923	0.3492	0.1439
0.9104	5.71	40	0.9589	0.1818	0.3421	0.3596	nan	0.0145	0.3590	0.7644	0.2304	0.0144	0.3436	0.3778	0.1731
0.7567	8.57	60	0.7761	0.2203	0.3739	0.3852	nan	0.0808	0.3882	0.6422	0.3844	0.0803	0.3778	0.4073	0.2360
0.7035	11.43	80	0.7442	0.2729	0.4718	0.4941	nan	0.2145	0.5077	0.8370	0.3279	0.2134	0.4817	0.4073	0.2619
0.5781	14.29	100	0.6260	0.2876	0.4446	0.4279	nan	0.4235	0.3777	0.5787	0.3986	0.3683	0.3761	0.4063	0.2873
0.5438	17.14	120	0.5559	0.3877	0.5412	0.5761	nan	0.5803	0.6504	0.5190	0.4149	0.5671	0.6193	0.4171	0.3352
0.5198	20.0	140	0.5617	0.3724	0.5617	0.5335	nan	0.6661	0.4532	0.7059	0.4216	0.6419	0.4532	0.4129	0.3540
0.4435	22.86	160	0.5393	0.4160	0.6198	0.6126	nan	0.7555	0.5832	0.6962	0.4442	0.7000	0.5705	0.4873	0.3221
0.5002	25.71	180	0.5126	0.4094	0.6080	0.6043	nan	0.6854	0.5833	0.6945	0.4687	0.6771	0.5761	0.4762	0.3176
0.4142	28.57	200	0.4874	0.3503	0.5361	0.4949	nan	0.6967	0.3895	0.6436	0.4147	0.6287	0.3895	0.4106	0.3228
0.3092	31.43	220	0.4819	0.3857	0.6001	0.5534	nan	0.7296	0.4267	0.8020	0.4423	0.7157	0.4267	0.4266	0.3595
0.2895	34.29	240	0.4969	0.3983	0.6220	0.5809	nan	0.7353	0.4689	0.8050	0.4787	0.7265	0.4677	0.4474	0.3498
0.3046	37.14	260	0.4767	0.4248	0.6412	0.6115	nan	0.7853	0.5270	0.7814	0.4711	0.7712	0.5199	0.4587	0.3742
0.3514	40.0	280	0.4531	0.3978	0.5989	0.5767	nan	0.7112	0.5082	0.7478	0.4282	0.6979	0.5024	0.4353	0.3537
0.2891	42.86	300	0.4629	0.3842	0.5885	0.5488	nan	0.7046	0.4397	0.7693	0.4403	0.6982	0.4366	0.4237	0.3623
0.2512	45.71	320	0.4584	0.3783	0.5794	0.5357	nan	0.7144	0.4199	0.7390	0.4443	0.7016	0.4196	0.4134	0.3568
0.2695	48.57	340	0.4592	0.3826	0.5892	0.5467	nan	0.7143	0.4323	0.7629	0.4472	0.7065	0.4317	0.4223	0.3527

Framework versions

Transformers 4.35.2
Pytorch 2.1.0+cu118
Datasets 2.15.0
Tokenizers 0.15.0

Task: Image Segmentation
Library: transformers
Type: segformer
License: other
Created: Dec 5, 2023
Updated: Dec 5, 2023