Segformer_finetune

saad7489/Segformer_finetune

Segformer_finetune is a image segmentation model from saad7489. Use it for the image segmentation task on the model card, and read the license before you ship it in a product. The card lists the license as other.

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

Downloads · 30 days

26% of all-time downloads

All-time downloads

Public

Parameters

3.7M

14.9 MB on disk

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.safetensors14.9 MB · 100%

At a glance

Task: Image Segmentation
License: other
Model type: segformer
Access: Public
Created: Aug 28, 2024
Updated: Aug 28, 2024
SHA: a8bd1f66

Base models

nvidia/mit-b0

Model card

From the Hugging Face model README

Segformer_finetune

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

Loss: 0.8216
Mean Iou: 0.4681
Mean Accuracy: 0.8710
Overall Accuracy: 0.9550
Accuracy Bkg: 0.9636
Accuracy Knife: 0.7783
Accuracy Gun: nan
Iou Bkg: 0.9533
Iou Knife: 0.4511
Iou Gun: 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: 5e-05
train_batch_size: 30
eval_batch_size: 30
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 Bkg	Accuracy Knife	Accuracy Gun	Iou Bkg	Iou Knife	Iou Gun
0.759	6.6667	20	0.8216	0.4681	0.8710	0.9550	0.9636	0.7783	nan	0.9533	0.4511	0.0

Framework versions

Transformers 4.42.4
Pytorch 2.4.0+cu121
Datasets 2.21.0
Tokenizers 0.19.1

Task: Image Segmentation
Type: segformer
License: other
Created: Aug 28, 2024
Updated: Aug 28, 2024