Edit model card

segformer-b0-finetuned-segments-construction-1

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

  • Loss: 0.2796
  • Mean Iou: 0.3218
  • Mean Accuracy: 0.5305
  • Overall Accuracy: 0.9276
  • Accuracy Unlabeled: nan
  • Accuracy Ruler: 0.8954
  • Accuracy Socket: 0.0
  • Accuracy Wall: 0.9644
  • Accuracy Window: nan
  • Accuracy Heater: nan
  • Accuracy Floor: 0.6710
  • Accuracy Ceiling: 0.0
  • Accuracy Skirting: nan
  • Accuracy Door: 0.6525
  • Accuracy Light: nan
  • Iou Unlabeled: nan
  • Iou Ruler: 0.7222
  • Iou Socket: 0.0
  • Iou Wall: 0.9553
  • Iou Window: 0.0
  • Iou Heater: nan
  • Iou Floor: 0.2630
  • Iou Ceiling: 0.0
  • Iou Skirting: 0.0
  • Iou Door: 0.6342
  • Iou Light: 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: 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 Ruler Accuracy Socket Accuracy Wall Accuracy Window Accuracy Heater Accuracy Floor Accuracy Ceiling Accuracy Skirting Accuracy Door Accuracy Light Iou Unlabeled Iou Ruler Iou Socket Iou Wall Iou Window Iou Heater Iou Floor Iou Ceiling Iou Skirting Iou Door Iou Light
1.8126 1.43 20 2.1233 0.1955 0.5160 0.8448 nan 0.8191 0.0 0.8868 nan nan 0.9618 0.0 nan 0.4281 nan 0.0 0.5555 0.0 0.8845 0.0 0.0 0.2971 0.0 0.0 0.4135 0.0
1.905 2.86 40 1.3611 0.1827 0.4921 0.8505 nan 0.9275 0.0 0.9139 nan nan 0.9627 0.0 nan 0.1484 nan nan 0.5404 0.0 0.9095 0.0 0.0 0.2289 0.0 0.0 0.1484 0.0
1.1072 4.29 60 1.0502 0.2327 0.5517 0.8903 nan 0.9108 0.0 0.9266 nan nan 0.9367 0.0 nan 0.5360 nan nan 0.5301 0.0 0.9206 0.0 0.0 0.3475 0.0 0.0 0.5284 0.0
1.0076 5.71 80 0.8802 0.2744 0.5609 0.9089 nan 0.8208 0.0 0.9410 nan nan 0.9532 0.0 nan 0.6505 nan nan 0.5500 0.0 0.9277 0.0 0.0 0.3688 0.0 0.0 0.6227 nan
1.5533 7.14 100 0.8991 0.2846 0.5514 0.8878 nan 0.8918 0.0 0.9243 nan nan 0.9591 0.0 nan 0.5332 nan nan 0.5262 0.0 0.9169 0.0 nan 0.3209 0.0 0.0 0.5132 nan
0.9912 8.57 120 0.9340 0.2891 0.5652 0.8854 nan 0.9478 0.0 0.9151 nan nan 0.9438 0.0 nan 0.5844 nan nan 0.5059 0.0 0.9098 0.0 nan 0.3424 0.0 0.0 0.5544 nan
0.784 10.0 140 0.7017 0.3140 0.5984 0.9173 nan 0.9136 0.0 0.9305 nan nan 0.8971 0.0 nan 0.8493 nan nan 0.5324 0.0 0.9224 0.0 0.0 0.5805 0.0 0.0 0.7909 nan
0.5636 11.43 160 0.6925 0.3573 0.5978 0.9280 nan 0.8714 0.0 0.9412 nan nan 0.8868 0.0 nan 0.8876 nan nan 0.5701 0.0 0.9308 0.0 nan 0.5638 0.0 0.0 0.7935 nan
1.0692 12.86 180 0.7313 0.2931 0.5724 0.8981 nan 0.9587 0.0 0.9231 nan nan 0.8880 0.0 nan 0.6647 nan nan 0.4988 0.0 0.9182 0.0 nan 0.3342 0.0 0.0 0.5932 nan
0.7603 14.29 200 0.6907 0.2577 0.5744 0.9001 nan 0.9619 0.0 0.9251 nan nan 0.8930 0.0 nan 0.6661 nan nan 0.4939 0.0 0.9208 0.0 0.0 0.3219 0.0 0.0 0.5824 nan
0.9509 15.71 220 0.5110 0.3682 0.6069 0.9324 nan 0.9355 0.0 0.9417 nan nan 0.8453 0.0 nan 0.9191 nan nan 0.5671 0.0 0.9334 0.0 nan 0.6050 0.0 0.0 0.8403 nan
0.4254 17.14 240 0.5925 0.2961 0.5629 0.9023 nan 0.9646 0.0 0.9295 nan nan 0.8261 0.0 nan 0.6569 nan nan 0.5302 0.0 0.9243 0.0 nan 0.3138 0.0 0.0 0.6009 nan
0.3839 18.57 260 0.4226 0.3537 0.5479 0.9367 nan 0.9108 0.0 0.9540 nan nan 0.5102 0.0 nan 0.9124 nan nan 0.6643 0.0 0.9426 0.0 nan 0.3868 0.0 0.0 0.8361 nan
0.7441 20.0 280 0.5084 0.3533 0.5993 0.9277 nan 0.9691 0.0 0.9391 nan nan 0.8075 0.0 nan 0.8801 nan nan 0.5527 0.0 0.9333 0.0 nan 0.5197 0.0 0.0 0.8208 nan
0.4374 21.43 300 0.4683 0.3038 0.5549 0.9173 nan 0.9662 0.0 0.9480 nan nan 0.7594 0.0 nan 0.6558 nan nan 0.6024 0.0 0.9419 0.0 nan 0.2804 0.0 0.0 0.6056 nan
0.6224 22.86 320 0.4100 0.3810 0.5960 0.9374 nan 0.9704 0.0 0.9460 nan nan 0.7131 0.0 nan 0.9467 nan nan 0.5986 0.0 0.9401 0.0 nan 0.6197 0.0 0.0 0.8898 nan
0.4473 24.29 340 0.3933 0.3368 0.5431 0.9336 nan 0.9212 0.0 0.9620 nan nan 0.6197 0.0 nan 0.7556 nan nan 0.7221 0.0 0.9521 0.0 nan 0.2958 0.0 0.0 0.7245 nan
0.3364 25.71 360 0.4336 0.2976 0.5125 0.9134 nan 0.9408 0.0 0.9544 nan nan 0.6075 0.0 nan 0.5721 nan nan 0.6918 0.0 0.9481 0.0 nan 0.1998 0.0 0.0 0.5411 nan
0.281 27.14 380 0.3795 0.3689 0.5760 0.9420 nan 0.9250 0.0 0.9589 nan nan 0.6859 0.0 nan 0.8863 nan nan 0.7108 0.0 0.9518 0.0 nan 0.4576 0.0 0.0 0.8305 nan
0.3198 28.57 400 0.4023 0.3158 0.5143 0.9238 nan 0.9120 0.0 0.9610 nan nan 0.5580 0.0 nan 0.6550 nan nan 0.7210 0.0 0.9519 0.0 nan 0.2238 0.0 0.0 0.6293 nan
0.4624 30.0 420 0.3565 0.3770 0.5774 0.9475 nan 0.9408 0.0 0.9613 nan nan 0.6287 0.0 nan 0.9337 nan nan 0.6855 0.0 0.9539 0.0 nan 0.4943 0.0 0.0 0.8827 nan
0.2356 31.43 440 0.3940 0.3100 0.5349 0.9221 nan 0.9268 0.0 0.9602 nan nan 0.7187 0.0 nan 0.6040 nan nan 0.7005 0.0 0.9536 0.0 nan 0.2474 0.0 0.0 0.5781 nan
0.3931 32.86 460 0.3516 0.3162 0.5570 0.9258 nan 0.9338 0.0 0.9598 nan nan 0.8124 0.0 nan 0.6362 nan nan 0.6824 0.0 0.9542 0.0 nan 0.2888 0.0 0.0 0.6040 nan
0.2431 34.29 480 0.4011 0.2955 0.5291 0.9138 nan 0.9242 0.0 0.9583 nan nan 0.7864 0.0 nan 0.5058 nan nan 0.6954 0.0 0.9520 0.0 nan 0.2331 0.0 0.0 0.4832 nan
0.2131 35.71 500 0.2847 0.3764 0.5613 0.9487 nan 0.8877 0.0 0.9679 nan nan 0.6103 0.0 nan 0.9020 nan nan 0.7330 0.0 0.9571 0.0 nan 0.4539 0.0 0.0 0.8669 nan
0.4151 37.14 520 0.3176 0.3186 0.5239 0.9256 nan 0.8930 0.0 0.9640 nan nan 0.6505 0.0 nan 0.6356 nan nan 0.7251 0.0 0.9544 0.0 nan 0.2507 0.0 0.0 0.6187 nan
0.2408 38.57 540 0.3267 0.3071 0.5361 0.9208 nan 0.9264 0.0 0.9600 nan nan 0.7441 0.0 nan 0.5859 nan nan 0.6868 0.0 0.9538 0.0 nan 0.2526 0.0 0.0 0.5635 nan
0.2274 40.0 560 0.2875 0.3396 0.5471 0.9349 nan 0.9098 0.0 0.9626 nan nan 0.6456 0.0 nan 0.7649 nan nan 0.7018 0.0 0.9547 0.0 nan 0.3216 0.0 0.0 0.7387 nan
0.2452 41.43 580 0.2998 0.3181 0.5357 0.9279 nan 0.9089 0.0 0.9642 nan nan 0.6932 0.0 nan 0.6480 nan nan 0.7057 0.0 0.9562 0.0 nan 0.2578 0.0 0.0 0.6252 nan
0.2922 42.86 600 0.2957 0.3131 0.5246 0.9255 nan 0.9056 0.0 0.9643 nan nan 0.6535 0.0 nan 0.6242 nan nan 0.7103 0.0 0.9563 0.0 nan 0.2347 0.0 0.0 0.6037 nan
0.3704 44.29 620 0.3290 0.3172 0.5429 0.9247 nan 0.9246 0.0 0.9583 nan nan 0.7123 0.0 nan 0.6621 nan nan 0.6856 0.0 0.9527 0.0 nan 0.2707 0.0 0.0 0.6286 nan
0.2482 45.71 640 0.2995 0.3251 0.5368 0.9276 nan 0.9018 0.0 0.9617 nan nan 0.6795 0.0 nan 0.6779 nan nan 0.7154 0.0 0.9538 0.0 nan 0.2790 0.0 0.0 0.6528 nan
0.2798 47.14 660 0.2808 0.3323 0.5374 0.9319 nan 0.8938 0.0 0.9644 nan nan 0.6554 0.0 nan 0.7110 nan nan 0.7218 0.0 0.9554 0.0 nan 0.2919 0.0 0.0 0.6894 nan
0.2746 48.57 680 0.2695 0.3265 0.5341 0.9299 nan 0.8947 0.0 0.9642 nan nan 0.6597 0.0 nan 0.6861 nan nan 0.7198 0.0 0.9554 0.0 nan 0.2735 0.0 0.0 0.6633 nan
0.2169 50.0 700 0.2796 0.3218 0.5305 0.9276 nan 0.8954 0.0 0.9644 nan nan 0.6710 0.0 nan 0.6525 nan nan 0.7222 0.0 0.9553 0.0 nan 0.2630 0.0 0.0 0.6342 nan

Framework versions

  • Transformers 4.25.1
  • Pytorch 1.13.1
  • Datasets 2.10.1
  • Tokenizers 0.13.0.dev0
Downloads last month
16
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.