Saltar al contenido
ModelPublicadoExterno
Browse files

Segformer B0 FT Uff Roi Segmentation All5 Vc1 Pc0 Cw1 E50 260717 2234

This model is a fine-tuned version of nvidia/mit-b0 on the LeninGF/uff-thermography-multiview-segmentation dataset. It achieves the following results on the evaluation set: Loss: 0.0878 Mean Iou: 0.9213 Mean...

TipoOtherModelo basenvidia/mit-b0VersiónRevision 26de674Bcalidad

Fuente del modelo

Tipo
Other
Tamaño del archivo
14,2 MB
Modelo base
nvidia/mit-b0
Versión
vRevision 26de674
Actualizado
10 oct 2026
Descargas
11
Fuente
Hugging Face

Descripción de la fuente

segformer-b0-FT-uff-roi-segmentation-all5-vc1-pc0-cw1-e50-260717-2234

This model is a fine-tuned version of nvidia/mit-b0 on the LeninGF/uff-thermography-multiview-segmentation dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0878
  • Mean Iou: 0.9213
  • Mean Accuracy: 0.9619
  • Overall Accuracy: 0.9605
  • Accuracy Background: 0.9553
  • Accuracy Roi: 0.9685
  • Iou Background: 0.9359
  • Iou Roi: 0.9067
  • Roi Precision: 0.9343
  • Roi Recall: 0.9685
  • Roi F1: 0.9511
  • Roi Dice: 0.9511
  • Roi Specificity: 0.9553

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: 1e-05
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 42
  • optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: linear
  • num_epochs: 50

Training results

Training LossEpochStepValidation LossMean IouMean AccuracyOverall AccuracyAccuracy BackgroundAccuracy RoiIou BackgroundIou RoiRoi PrecisionRoi RecallRoi F1Roi DiceRoi Specificity
0.32741.01240.31630.69520.84670.82060.72090.97250.70810.68220.69570.97250.81110.81110.7209
0.25832.02480.23820.79160.89830.88530.83560.96110.81480.76850.79320.96110.86910.86910.8356
0.21543.03720.21000.80520.90830.89340.83660.98000.82570.78460.79730.98000.87930.87930.8366
0.22064.04960.18780.84130.92540.91550.87750.97330.86240.82020.83910.97330.90120.90120.8775
0.20055.06200.16020.87690.94010.93630.92200.95820.89740.85640.88960.95820.92260.92260.9220
0.16726.07440.14960.87510.94110.93510.91210.97010.89460.85560.87870.97010.92220.92220.9121
0.15847.08680.14570.87330.94130.93400.90620.97650.89240.85430.87220.97650.92140.92140.9062
0.16508.09920.13490.88470.94480.94060.92460.96510.90390.86560.89360.96510.92800.92800.9246
0.12429.011160.13440.88220.94570.93900.91320.97820.90040.86400.88090.97820.92700.92700.9132
0.142310.012400.12510.89230.94910.94480.92840.96980.91030.87430.89880.96980.93290.93290.9284
0.135911.013640.12160.89480.94990.94620.93190.96790.91270.87690.90310.96790.93440.93440.9319
0.141512.014880.12070.89310.95030.94510.92510.97550.91050.87560.89530.97550.93370.93370.9251
0.107613.016120.12220.89090.94940.94390.92270.97610.90850.87330.89230.97610.93230.93230.9227
0.129914.017360.11110.90290.95380.95060.93820.96950.91980.88600.91140.96950.93960.93960.9382
0.101815.018600.11280.89910.95320.94840.93010.97630.91590.88230.90170.97630.93750.93750.9301
0.119516.019840.11770.89460.95100.94600.92690.97500.91200.87730.89740.97500.93460.93460.9269
0.109217.021080.10870.90190.95440.95000.93300.97590.91840.88540.90520.97590.93920.93920.9330
0.105118.022320.10520.90580.95540.95220.94000.97080.92230.88940.91380.97080.94150.94150.9400
0.094519.023560.10760.90370.95440.95100.93810.97060.92040.88700.91150.97060.94010.94010.9381
0.109320.024800.10210.90800.95750.95330.93720.97770.92370.89230.91080.97770.94310.94310.9372
0.127021.026040.10020.91370.95780.95650.95130.96440.92960.89780.92850.96440.94610.94610.9513
0.095322.027280.09730.91410.95850.95670.94980.96730.92980.89850.92670.96730.94650.94650.9498
0.122623.028520.09760.91580.95780.95770.95740.95810.93180.89970.93660.95810.94720.94720.9574
0.080624.029760.09650.91290.95920.95590.94320.97530.92810.89760.91850.97530.94600.94600.9432
0.087725.031000.09640.91320.95870.95610.94640.97100.92870.89760.92230.97100.94610.94610.9464
0.098226.032240.10040.90630.95740.95230.93280.98190.92190.89070.90560.98190.94220.94220.9328
0.104927.033480.09740.91170.95850.95530.94330.97370.92730.89620.91840.97370.94530.94530.9433
0.075828.034720.09380.91430.96000.95670.94400.97590.92930.89920.91960.97590.94690.94690.9440
0.098929.035960.09220.91790.96020.95870.95330.96700.93310.90280.93140.96700.94890.94890.9533
0.077930.037200.09290.91660.96000.95800.95060.96930.93180.90140.92790.96930.94820.94820.9506
0.066131.038440.09020.91900.96120.95920.95190.97040.93380.90410.92970.97040.94970.94970.9519
0.070132.039680.09530.91300.95910.95600.94400.97430.92830.89770.91940.97430.94610.94610.9440
0.095233.040920.09240.91780.95930.95870.95630.96240.93330.90230.93530.96240.94860.94860.9563
0.083434.042160.09150.91600.96040.95760.94700.97370.93100.90100.92340.97370.94790.94790.9470
0.087735.043400.08970.91860.96110.95900.95130.97080.93340.90370.92900.97080.94940.94940.9513
0.065036.044640.09000.91890.96100.95920.95250.96950.93380.90410.93060.96950.94960.94960.9525
0.081137.045880.09080.91810.96060.95880.95160.96970.93300.90310.92930.96970.94910.94910.9516
0.082338.047120.08910.92030.96140.96000.95440.96840.93500.90550.93300.96840.95040.95040.9544
0.063139.048360.09190.91560.96040.95740.94580.97510.93050.90060.92180.97510.94770.94770.9458
0.073140.049600.08920.91950.96120.95950.95300.96950.93430.90470.93120.96950.94990.94990.9530
0.080441.050840.08830.92070.96170.96020.95450.96880.93540.90600.93320.96880.95070.95070.9545
0.076542.052080.09040.91840.96090.95890.95160.97010.93330.90340.92930.97010.94930.94930.9516
0.070643.053320.08860.92100.96140.96040.95640.96640.93580.90610.93570.96640.95080.95080.9564
0.075344.054560.08900.91980.96120.9597...

Fuentes

1 fuente

Download files

Artefactos del modelo

2 artefactos

best-model/model.safetensors

safetensors · 14,2 MB · SHA-256 e7868e9ec6a2…ac7f · Hugging Face

Download file

model.safetensors

safetensors · 14,2 MB · SHA-256 e7868e9ec6a2…ac7f · Hugging Face

Download file

Extractos de fuentes

3 extractos

--- library_name: transformers license: other base_model: nvidia/mit-b0 tags: - vision - image-segmentation - medical-imaging - thermography - multiview - breast-cancer - roi-segmentation - generated_from_trainer model-index: - name: segformer-b0-FT-uff-roi-segmentation-all5-vc1-pc0-cw1-e50-260717-2234 results: [] --- <!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. --> # segformer-b0-FT-uff-roi-segmentation-all5-vc1-pc0-cw1-e50-260717-2234 This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the LeninGF/uff-thermography-multiview-segmentation dataset. It achieves the following results on the evaluation set: - Loss: 0.0878 - Mean Iou: 0.9213 - Mean Accuracy: 0.9619 - Overall Accuracy: 0.9605 - Accuracy Background: 0.9553 - Accuracy Roi: 0.9685 - Iou Background: 0.9359 - Iou Roi: 0.9067 - Roi Precision: 0.9343 - Roi Recall: 0.9685 - Roi F1: 0.9511 - Roi Dice: 0.9511 - Roi Specificity: 0.9553 ## 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: 1e-05 - train_batch_size: 4 - eval_batch_size: 4 - seed: 42 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments - lr_scheduler_type: linear - num_epochs: 50 ### Training results | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Background | Accuracy Roi | Iou Background | Iou Roi | Roi Precision | Roi Recall | Roi F1 | Roi Dice | Roi Specificity | |:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:-------------------:|:------------:|:--------------:|:-------:|:-------------:|:----------:|:------:|:--------:|:---------------:| | 0.3274 | 1.0 | 124 | 0.3163 | 0.6952 | 0.8467 | 0.8206 | 0.7209 | 0.9725 | 0.7081 | 0.6822 | 0.6957 | 0.9725 | 0.8111 | 0.8111 | 0.7209 | | 0.2583 | 2.0 | 248 | 0.2382 | 0.7916 | 0.8983 | 0.8853 | 0.8356 | 0.9611 | 0.8148 | 0.7685 | 0.7932 | 0.9611 | 0.8691 | 0.8691 | 0.8356 | | 0.2154 | 3.0 | 372...

Source context: 11 downloads · 0 likes · Pipeline image-segmentation · Library transformers · Repo LeninGF/segformer-b0-FT-uff-roi-segmentation-all5-vc1-pc0-cw1-e50-260717-2234

Source context: 21 downloads · 0 likes · Pipeline image-segmentation · Library transformers · Repo LeninGF/segformer-b0-FT-uff-roi-segmentation-all5-vc1-pc0-cw1-e50-260717-2234