Edit model card

segformer-b0-drone-real

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

  • Loss: 2.5066
  • Mean Iou: 0.0277
  • Mean Accuracy: 0.0561
  • Overall Accuracy: 0.3272
  • Per Category Iou: [nan, 0.3359579661339679, 0.0, 0.20517716653585355, 0.003702068923287315, 0.0, 0.0, 0.0, 0.006312134342948017, 0.058787282412541476, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
  • Per Category Accuracy: [nan, 0.6093986043791881, 0.0, 0.4105641817321649, 0.003921086446309355, 0.0, 0.0, 0.0, 0.00706877745310955, 0.20348775859354865, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]

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: 16
  • eval_batch_size: 16
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 20

Training results

Training Loss Epoch Step Validation Loss Mean Iou Mean Accuracy Overall Accuracy Per Category Iou Per Category Accuracy
2.8285 2.0 40 2.8542 0.0229 0.0536 0.2211 [nan, 0.23060686675928652, 3.3649639948852546e-05, 0.16273692278074856, 0.013325250843685254, 4.063223761732559e-05, 0.0, 0.0, 0.035811860573766785, 0.06198642655834834, 0.0, 0.0, 6.128953174797744e-05, 0.0, 0.0, 0.0, 5.1148278860416345e-05, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.33385338422824606, 3.373169528473065e-05, 0.299629413956809, 0.014924989967585504, 4.09158863797309e-05, 0.0, 0.0, 0.06156816279101685, 0.46948081490592114, 0.0, 0.0, 7.101768340316739e-05, 0.0, 0.0, 0.0, 5.163155720776539e-05, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
2.6567 4.0 80 2.6894 0.0262 0.0563 0.2866 [nan, 0.30624516924744655, 0.0, 0.18452743224435408, 0.004815305953306897, 0.0, 0.0, 0.0, 0.013916000286575587, 0.06596235864669749, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.49410241551371004, 0.0, 0.3789267923602906, 0.005102681931249924, 0.0, 0.0, 0.0, 0.016502745882809654, 0.3439400151190408, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
2.3286 6.0 120 2.5923 0.0267 0.0560 0.3328 [nan, 0.35729772739055154, 0.0, 0.1539404924138176, 0.0035934817901112264, 0.0, 0.0, 0.0, 0.005766957703680072, 0.06586525335295575, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.7280486819078453, 0.0, 0.23220288304265976, 0.003859608350752127, 0.0, 0.0, 0.0, 0.006166880921555261, 0.2614486947285752, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
2.3836 8.0 160 2.5890 0.0275 0.0575 0.3244 [nan, 0.349601471481378, 0.0, 0.17496917223449057, 0.0044570694472161245, 0.0, 0.0, 0.0, 0.007437757319012064, 0.0682080200846761, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.6614103864255038, 0.0, 0.29170107802854334, 0.004828395043379214, 0.0, 0.0, 0.0, 0.008074588150255963, 0.29956113264795986, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
2.2849 10.0 200 2.5525 0.0283 0.0569 0.3388 [nan, 0.3351423574874955, 0.0, 0.21293393590619594, 0.002350142484175345, 0.0, 0.0, 0.0, 0.005453996034797477, 0.06749403988523864, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.6106816065491506, 0.0, 0.4703984171366385, 0.002485471577527932, 0.0, 0.0, 0.0, 0.0058035081160884615, 0.16273986773434082, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
2.1539 12.0 240 2.5352 0.0289 0.0579 0.3431 [nan, 0.34790380657207765, 0.0, 0.21813897765519252, 0.0039179488370322925, 0.0, 0.0, 0.0, 0.004069615148127275, 0.06184727957512016, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.6213138456628569, 0.0, 0.4695635471053968, 0.004103493982578053, 0.0, 0.0, 0.0, 0.004355898828122703, 0.17384734612397268, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
2.1894 14.0 280 2.5923 0.0271 0.0567 0.3056 [nan, 0.3019300415296601, 0.0, 0.2108717486241356, 0.003421017543191299, 0.0, 0.0, 0.0, 0.012217783066499338, 0.06721066611554333, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.4912292097745947, 0.0, 0.49141307925385636, 0.0036400437237620272, 0.0, 0.0, 0.0, 0.01400423107111977, 0.2475275870599001, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
1.9499 16.0 320 2.5169 0.0282 0.0569 0.3320 [nan, 0.33589786653498954, 0.0, 0.20789931305652776, 0.004207056605876958, 0.0, 0.0, 0.0, 0.0075080990386678535, 0.06424320004285503, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.613616488775929, 0.0, 0.4253884604222434, 0.0044872253922099824, 0.0, 0.0, 0.0, 0.008369991941750555, 0.19915876549744932, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
2.115 18.0 360 2.5153 0.0277 0.0562 0.3256 [nan, 0.3342799309383031, 0.0, 0.2066227292961174, 0.0034518271833141106, 0.0, 0.0, 0.0, 0.006799912048482982, 0.05915124911379064, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.5983876453859164, 0.0, 0.4204451510267332, 0.0036636891451301916, 0.0, 0.0, 0.0, 0.007619718595211191, 0.20718275529815128, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]
2.0119 20.0 400 2.5066 0.0277 0.0561 0.3272 [nan, 0.3359579661339679, 0.0, 0.20517716653585355, 0.003702068923287315, 0.0, 0.0, 0.0, 0.006312134342948017, 0.058787282412541476, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan] [nan, 0.6093986043791881, 0.0, 0.4105641817321649, 0.003921086446309355, 0.0, 0.0, 0.0, 0.00706877745310955, 0.20348775859354865, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, nan]

Framework versions

  • Transformers 4.25.1
  • Pytorch 1.12.1+cu113
  • Datasets 2.7.1
  • Tokenizers 0.13.2
Downloads last month
5
Inference API
Unable to determine this model’s pipeline type. Check the docs .