CoolFace
Modelpublic

peldrak/segformer-finetuned-coastal

sourceHugging Faceotherupdated 3y agoView on Hugging Face
0likes8downloads
Model Card

<!-- 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-finetuned-coastal

This model is a fine-tuned version of nvidia/segformer-b0-finetuned-ade-512-512 on the peldrak/coastal3 dataset. It achieves the following results on the evaluation set:

  • —Loss: 1.7456
  • —Mean Iou: 0.0841
  • —Mean Accuracy: 0.1629
  • —Overall Accuracy: 0.3309
  • —Accuracy Water: 0.2728
  • —Accuracy Whitewater: 0.0
  • —Accuracy Sediment: 0.0194
  • —Accuracy Other Natural Terrain: 0.0
  • —Accuracy Vegetation: 0.7325
  • —Accuracy Development: 0.0055
  • —Accuracy Unknown: 0.1102
  • —Iou Water: 0.1767
  • —Iou Whitewater: 0.0
  • —Iou Sediment: 0.0157
  • —Iou Other Natural Terrain: 0.0
  • —Iou Vegetation: 0.3002
  • —Iou Development: 0.0052
  • —Iou Unknown: 0.0908

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
  • —trainbatchsize: 8
  • —evalbatchsize: 8
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —num_epochs: 5

Training results

Training LossEpochStepValidation LossMean IouMean AccuracyOverall AccuracyAccuracy WaterAccuracy WhitewaterAccuracy SedimentAccuracy Other Natural TerrainAccuracy VegetationAccuracy DevelopmentAccuracy UnknownIou WaterIou WhitewaterIou SedimentIou Other Natural TerrainIou VegetationIou DevelopmentIou Unknown
1.84290.12201.87420.06230.14320.21980.51720.00440.31370.00500.11650.00190.04350.25140.00370.05210.00340.08890.00190.0348
1.66230.25401.82970.07310.15720.25360.43490.00070.33360.00100.31580.00110.01350.24620.00060.05690.00090.19400.00110.0120
1.61730.37601.78560.08120.15850.30800.31510.00070.11790.00050.63390.00120.04050.20560.00070.03570.00050.28590.00120.0388
1.50780.49801.78050.07640.15370.30380.27920.00.10360.00000.66390.00030.02900.18390.00.03380.00000.28950.00030.0274
1.39270.621001.75490.07250.15210.32270.25890.00.02220.00010.76430.00050.01870.17030.00.01330.00010.30550.00050.0179
1.60240.741201.73480.08350.15980.32710.35140.00.02040.00010.65380.00050.09260.20210.00.01470.00010.28730.00050.0796
1.32870.861401.73520.07760.15560.32670.33090.00.00830.00.69100.00000.05910.19350.00.00760.00.28490.00000.0573
1.41530.991601.70240.07460.15670.33560.29470.00.00110.00.76830.00060.03200.18790.00.00110.00.30210.00060.0309
1.33341.111801.72620.07440.15670.33740.31300.00.00180.00.76200.00010.02030.19880.00.00170.00.30040.00010.0198
1.39561.232001.73040.08580.16220.33260.48380.00.01270.00.54320.00150.09440.23730.00.01120.00.27310.00150.0777
1.57761.362201.73000.07910.16220.34110.25810.00.00100.00.80120.00030.07480.17340.00.00100.00.31440.00030.0647
1.16561.482401.72480.08310.16570.34400.26870.00.00260.00.78790.00140.09950.17750.00.00250.00.31830.00140.0822
1.44291.62601.73080.07640.16160.34080.20910.00.00290.00.85180.00370.06370.15070.00.00280.00.32000.00360.0578
1.66491.732801.72820.07430.15640.33720.32610.00.00330.00.75140.00150.01280.19940.00.00310.00.30370.00150.0125
1.36341.853001.72160.08470.16530.34130.31960.00.02730.00.74020.00360.06650.20120.00.02040.00.31010.00350.0579
1.52241.983201.73430.08220.16260.34100.37930.00.02770.00.69850.00170.03110.22360.00.02100.00.30000.00160.0293
1.25272.13401.71490.07590.15590.33440.41080.00.01190.00.66290.00210.00340.22440.00.01050.00.29130.00210.0033
1.59312.223601.71700.08380.16190.33650.40610.00.00980.00.64140.00280.07300.22280.00.00900.00.28820.00270.0641
1.24342.353801.74370.08440.16190.33640.47340.00.01180.00.57850.00280.06670.23910.00.01050.00.27770.00270.0607
1.40712.474001.73160.08230.16390.34330.30540.00.00530.00.76330.00570.06740.19330.00.00510.00.31250.00540.0596
1.21772.594201.71950.08480.16570.34590.34420.00.00300.00.73150.01100.07050.20840.00.00290.00.31110.01020.0613
1.37242.724401.73590.08430.16600.34550.30910.00.00890.00.76200.00570.07610.19460.00.00840.00.31470.00540.0669
1.39732.844601.74690.08270.16170.33520.31530.00.01010.00.72310.01090.07240.19220.00.00880.00.30450.00990.0638
1.30982.964801.71930.08520.16580.34470.34120.00.00320.00.72400.00380.08870.20390.00.00310.00.30760.00370.0779
0.95453.095001.72560.08400.16270.33590.33660.00.00260.00.69590.00770.09600.20070.00.00250.00.29690.00720.0808
1.1763.215201.73340.08270.16160.33570.33960.00.00240.00.69630.00160.09150.19990.00.00240.00.29660.00150.0785
1.56223.335401.77900.06890.15280.33110.23930.00.00150.00.81860.00090.00910.16380.00.00150.00.30730.00090.0089
1.26733.465601.73390.08030.15850.33020.32280.00.01130.00.70350.00360.06860.19300.00.01040.00.29420.00340.0613
1.4183.585801.76480.07600.15630.33250.30740.00.00220.00.74160.00410.03860.19020.00.00220.00.29940.00380.0363
1.35783.76001.73380.08450.16190.33270.35480.00.01190.00.66930.00760.08980.20390.00.01080.00.29400.00700.0758
1.19913.836201.77110.07610.15460.32850.34730.00.00490.00.69060.00180.03790.19950.00.00460.00.29120.00170.0356
1.36993.956401.74210.08290.15950.33120.42900.00.00480.00.59940.00320.08040.22230.00.00450.00.28190.00310.0687
1.3084.076601.77690.07090.15180.32500.25120.00.00350.00.77840.00340.02600.16560.00.00330.00.29960.00330.0245
1.37464.26801.78110.07490.15380.32830.34970.00.00770.00.69530.00370.02000.20220.00.00700.00.29280.00350.0191
1.20854.327001.74010.08250.16000.33190.36320.00.01060.00.66630.00420.07590.20640.00.00960.00.29120.00400.0664
0.81194.447201.76380.07550.15390.32840.36600.00.00600.00.67840.00340.02370.20730.00.00530.00.29020.00330.0226
1.15474.577401.75810.07950.15730.32890.34100.00.01400.00.68790.00300.05500.19960.00.01160.00.29290.00290.0497
1.22294.697601.78170.07300.15500.32430.18610.00.01240.00.81980.00360.06310.13650.00.01030.00.30500.00340.0557
1.33324.817801.75800.07690.15630.32760.26560.00.00440.00.75240.00420.06770.17210.00.00410.00.29980.00400.0581
1.16684.948001.74560.08410.16290.33090.27280.00.01940.00.73250.00550.11020.17670.00.01570.00.30020.00520.0908

Framework versions

  • —Transformers 4.33.0
  • —Pytorch 2.0.0
  • —Datasets 2.1.0
  • —Tokenizers 0.13.3