File size: 10,266 Bytes
249fadb b8608fa 249fadb b8608fa 249fadb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
---
license: other
base_model: nvidia/mit-b0
tags:
- image-segmentation
- vision
- generated_from_trainer
model-index:
- name: segformer-finetuned-biofilm_MRCNNv1_train80_val20
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-finetuned-biofilm_MRCNNv1_train80_val20
This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the heroza/biofilm_MRCNNv1_train80_val20 dataset.
It achieves the following results on the evaluation set:
- Loss: 0.0785
- Mean Iou: 0.8728
- Mean Accuracy: 0.9695
- Overall Accuracy: 0.9773
- Accuracy Background: 0.9788
- Accuracy Biofilm: 0.9603
- Iou Background: 0.9755
- Iou Biofilm: 0.7702
## 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: 8
- eval_batch_size: 8
- seed: 1337
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: polynomial
- num_epochs: 50.0
### Training results
| Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Background | Accuracy Biofilm | Iou Background | Iou Biofilm |
|:-------------:|:-----:|:-----:|:---------------:|:--------:|:-------------:|:----------------:|:-------------------:|:----------------:|:--------------:|:-----------:|
| 0.1517 | 1.0 | 280 | 0.1319 | 0.8108 | 0.8974 | 0.9664 | 0.9794 | 0.8154 | 0.9641 | 0.6575 |
| 0.0728 | 2.0 | 560 | 0.0636 | 0.8857 | 0.9582 | 0.9807 | 0.9849 | 0.9314 | 0.9791 | 0.7923 |
| 0.0555 | 3.0 | 840 | 0.0526 | 0.8863 | 0.9417 | 0.9815 | 0.9889 | 0.8944 | 0.9800 | 0.7925 |
| 0.045 | 4.0 | 1120 | 0.0522 | 0.8763 | 0.9225 | 0.9802 | 0.9911 | 0.8539 | 0.9788 | 0.7738 |
| 0.048 | 5.0 | 1400 | 0.0415 | 0.9085 | 0.9742 | 0.9848 | 0.9868 | 0.9616 | 0.9835 | 0.8335 |
| 0.0425 | 6.0 | 1680 | 0.0468 | 0.9041 | 0.9790 | 0.9837 | 0.9846 | 0.9733 | 0.9824 | 0.8258 |
| 0.0438 | 7.0 | 1960 | 0.0461 | 0.9001 | 0.9717 | 0.9832 | 0.9853 | 0.9581 | 0.9818 | 0.8185 |
| 0.0406 | 8.0 | 2240 | 0.0488 | 0.8936 | 0.9701 | 0.9819 | 0.9841 | 0.9561 | 0.9804 | 0.8068 |
| 0.0354 | 9.0 | 2520 | 0.0434 | 0.9004 | 0.9648 | 0.9835 | 0.9870 | 0.9426 | 0.9821 | 0.8187 |
| 0.0411 | 10.0 | 2800 | 0.0467 | 0.9021 | 0.9680 | 0.9837 | 0.9867 | 0.9494 | 0.9824 | 0.8219 |
| 0.0385 | 11.0 | 3080 | 0.0846 | 0.8493 | 0.9697 | 0.9716 | 0.9720 | 0.9674 | 0.9692 | 0.7294 |
| 0.0327 | 12.0 | 3360 | 0.0574 | 0.8709 | 0.9643 | 0.9771 | 0.9795 | 0.9491 | 0.9753 | 0.7666 |
| 0.0333 | 13.0 | 3640 | 0.0654 | 0.8649 | 0.9645 | 0.9757 | 0.9778 | 0.9513 | 0.9737 | 0.7561 |
| 0.0356 | 14.0 | 3920 | 0.0823 | 0.8472 | 0.9703 | 0.9710 | 0.9712 | 0.9694 | 0.9686 | 0.7259 |
| 0.0277 | 15.0 | 4200 | 0.0657 | 0.8634 | 0.9761 | 0.9748 | 0.9745 | 0.9777 | 0.9727 | 0.7542 |
| 0.0328 | 16.0 | 4480 | 0.0575 | 0.8785 | 0.9668 | 0.9787 | 0.9810 | 0.9526 | 0.9770 | 0.7800 |
| 0.0362 | 17.0 | 4760 | 0.0595 | 0.8750 | 0.9696 | 0.9778 | 0.9794 | 0.9599 | 0.9760 | 0.7741 |
| 0.0301 | 18.0 | 5040 | 0.0610 | 0.8701 | 0.9755 | 0.9764 | 0.9766 | 0.9744 | 0.9744 | 0.7659 |
| 0.0284 | 19.0 | 5320 | 0.0562 | 0.8874 | 0.9748 | 0.9804 | 0.9814 | 0.9682 | 0.9787 | 0.7961 |
| 0.0345 | 20.0 | 5600 | 0.0612 | 0.8704 | 0.9696 | 0.9767 | 0.9781 | 0.9611 | 0.9748 | 0.7659 |
| 0.0292 | 21.0 | 5880 | 0.0689 | 0.8639 | 0.9728 | 0.9751 | 0.9755 | 0.9701 | 0.9730 | 0.7549 |
| 0.0287 | 22.0 | 6160 | 0.0559 | 0.8821 | 0.9657 | 0.9796 | 0.9822 | 0.9493 | 0.9779 | 0.7862 |
| 0.0307 | 23.0 | 6440 | 0.0683 | 0.8637 | 0.9716 | 0.9751 | 0.9757 | 0.9674 | 0.9730 | 0.7545 |
| 0.0303 | 24.0 | 6720 | 0.0692 | 0.8702 | 0.9627 | 0.9770 | 0.9797 | 0.9457 | 0.9752 | 0.7653 |
| 0.0257 | 25.0 | 7000 | 0.0594 | 0.8783 | 0.9709 | 0.9785 | 0.9799 | 0.9619 | 0.9767 | 0.7798 |
| 0.0341 | 26.0 | 7280 | 0.0762 | 0.8619 | 0.9746 | 0.9745 | 0.9745 | 0.9747 | 0.9723 | 0.7515 |
| 0.025 | 27.0 | 7560 | 0.0675 | 0.8696 | 0.9751 | 0.9763 | 0.9765 | 0.9736 | 0.9743 | 0.7649 |
| 0.0281 | 28.0 | 7840 | 0.0661 | 0.8641 | 0.9694 | 0.9753 | 0.9764 | 0.9625 | 0.9732 | 0.7550 |
| 0.0285 | 29.0 | 8120 | 0.0796 | 0.8592 | 0.9737 | 0.9739 | 0.9739 | 0.9736 | 0.9717 | 0.7467 |
| 0.0263 | 30.0 | 8400 | 0.0760 | 0.8627 | 0.9712 | 0.9749 | 0.9755 | 0.9668 | 0.9728 | 0.7527 |
| 0.0252 | 31.0 | 8680 | 0.0615 | 0.8800 | 0.9642 | 0.9792 | 0.9820 | 0.9464 | 0.9775 | 0.7825 |
| 0.0245 | 32.0 | 8960 | 0.0647 | 0.8665 | 0.9642 | 0.9761 | 0.9783 | 0.9501 | 0.9742 | 0.7589 |
| 0.0241 | 33.0 | 9240 | 0.0638 | 0.8749 | 0.9668 | 0.9779 | 0.9800 | 0.9535 | 0.9761 | 0.7736 |
| 0.0249 | 34.0 | 9520 | 0.0803 | 0.8610 | 0.9709 | 0.9744 | 0.9751 | 0.9667 | 0.9723 | 0.7497 |
| 0.0213 | 35.0 | 9800 | 0.0754 | 0.8687 | 0.9633 | 0.9767 | 0.9792 | 0.9474 | 0.9748 | 0.7627 |
| 0.0233 | 36.0 | 10080 | 0.0675 | 0.8743 | 0.9612 | 0.9780 | 0.9812 | 0.9411 | 0.9763 | 0.7723 |
| 0.0214 | 37.0 | 10360 | 0.0695 | 0.8758 | 0.9714 | 0.9779 | 0.9791 | 0.9637 | 0.9761 | 0.7755 |
| 0.0231 | 38.0 | 10640 | 0.0704 | 0.8695 | 0.9621 | 0.9769 | 0.9797 | 0.9444 | 0.9750 | 0.7640 |
| 0.0231 | 39.0 | 10920 | 0.0780 | 0.8636 | 0.9646 | 0.9754 | 0.9774 | 0.9518 | 0.9734 | 0.7539 |
| 0.0221 | 40.0 | 11200 | 0.0726 | 0.8709 | 0.9666 | 0.9770 | 0.9790 | 0.9542 | 0.9751 | 0.7666 |
| 0.0227 | 41.0 | 11480 | 0.0829 | 0.8618 | 0.9627 | 0.9751 | 0.9774 | 0.9480 | 0.9730 | 0.7505 |
| 0.0241 | 42.0 | 11760 | 0.0701 | 0.8763 | 0.9679 | 0.9782 | 0.9801 | 0.9557 | 0.9764 | 0.7762 |
| 0.0206 | 43.0 | 12040 | 0.0782 | 0.8666 | 0.9670 | 0.9760 | 0.9777 | 0.9563 | 0.9740 | 0.7593 |
| 0.023 | 44.0 | 12320 | 0.0809 | 0.8656 | 0.9654 | 0.9758 | 0.9778 | 0.9530 | 0.9739 | 0.7573 |
| 0.0223 | 45.0 | 12600 | 0.0805 | 0.8688 | 0.9660 | 0.9765 | 0.9785 | 0.9535 | 0.9746 | 0.7630 |
| 0.0224 | 46.0 | 12880 | 0.0748 | 0.8719 | 0.9682 | 0.9772 | 0.9789 | 0.9576 | 0.9753 | 0.7685 |
| 0.0233 | 47.0 | 13160 | 0.0796 | 0.8697 | 0.9669 | 0.9767 | 0.9786 | 0.9552 | 0.9748 | 0.7645 |
| 0.019 | 48.0 | 13440 | 0.0772 | 0.8729 | 0.9681 | 0.9774 | 0.9792 | 0.9569 | 0.9756 | 0.7703 |
| 0.0215 | 49.0 | 13720 | 0.0783 | 0.8720 | 0.9682 | 0.9772 | 0.9789 | 0.9575 | 0.9753 | 0.7686 |
| 0.0186 | 50.0 | 14000 | 0.0785 | 0.8728 | 0.9695 | 0.9773 | 0.9788 | 0.9603 | 0.9755 | 0.7702 |
### Framework versions
- Transformers 4.38.0.dev0
- Pytorch 2.0.0+cu117
- Datasets 2.14.4
- Tokenizers 0.15.1
|