CoolFace
Modelpublic

yeray142/finetune-instance-segmentation-mini-mask2former_augmentation_default

sourceHugging Faceotherupdated 2y agoView on Hugging Face
0likes11downloads
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. -->

finetune-instance-segmentation-mini-mask2formeraugmentationdefault

This model is a fine-tuned version of facebook/mask2former-swin-tiny-coco-instance on the jsalavedra/strawberry_disease dataset. It achieves the following results on the evaluation set:

  • Loss: 16.2566
  • Map: 0.6229
  • Map 50: 0.8172
  • Map 75: 0.6824
  • Map Small: 0.35
  • Map Medium: 0.401
  • Map Large: 0.7146
  • Mar 1: 0.4279
  • Mar 10: 0.7886
  • Mar 100: 0.831
  • Mar Small: 0.5
  • Mar Medium: 0.7108
  • Mar Large: 0.8802
  • Map Angular leafspot: 0.54
  • Mar 100 Angular leafspot: 0.8135
  • Map Anthracnose fruit rot: 0.405
  • Mar 100 Anthracnose fruit rot: 0.7118
  • Map Blossom blight: 0.7372
  • Mar 100 Blossom blight: 0.8159
  • Map Gray mold: 0.5767
  • Mar 100 Gray mold: 0.7648
  • Map Leaf spot: 0.8783
  • Mar 100 Leaf spot: 0.9416
  • Map Powdery mildew fruit: 0.5019
  • Mar 100 Powdery mildew fruit: 0.8833
  • Map Powdery mildew leaf: 0.7209
  • Mar 100 Powdery mildew leaf: 0.8863

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
  • trainbatchsize: 8
  • evalbatchsize: 8
  • seed: 42
  • gradientaccumulationsteps: 2
  • totaltrainbatch_size: 16
  • optimizer: Use OptimizerNames.ADAMWTORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • lrschedulertype: constant
  • num_epochs: 10.0
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossMapMap 50Map 75Map SmallMap MediumMap LargeMar 1Mar 10Mar 100Mar SmallMar MediumMar LargeMap Angular leafspotMar 100 Angular leafspotMap Anthracnose fruit rotMar 100 Anthracnose fruit rotMap Blossom blightMar 100 Blossom blightMap Gray moldMar 100 Gray moldMap Leaf spotMar 100 Leaf spotMap Powdery mildew fruitMar 100 Powdery mildew fruitMap Powdery mildew leafMar 100 Powdery mildew leaf
48.13831.09132.74410.0720.09440.07890.00080.0280.12180.14440.28150.35830.20.19770.42770.00970.49620.0010.05290.00250.07270.05270.22220.18180.84010.00020.050.25570.7737
28.13212.018226.82570.22030.28910.23950.12670.11860.26170.29870.53510.59130.40.37260.66490.02030.59810.00490.17060.17690.650.19610.63240.63550.90820.00870.34440.49980.8353
23.61923.027323.09340.31410.4080.34230.21330.1950.38340.36210.65160.69710.40.42910.77970.04570.72880.01750.40590.48730.72050.29840.71020.75750.92490.02570.54440.56660.8451
20.60544.036421.18370.3980.53030.4340.21150.2370.49980.38520.70620.75680.50.56240.83060.2640.73650.03680.58820.53860.750.4340.72870.79650.93310.1070.70560.60880.8557
19.42885.045519.91110.43660.59240.46860.17580.2560.52690.40050.72950.78870.40.59170.85580.31570.77880.06790.64710.57150.76360.47580.73890.81070.93150.18250.80.6320.8608
17.88676.054618.93020.50640.68340.54570.350.32320.56930.3950.74620.80060.450.64270.85830.34720.77310.18020.70.65160.79090.51930.73980.82090.93620.35840.80.66710.8639
16.99857.063718.26920.54580.72660.60360.26670.3530.61380.41450.77010.81420.40.64880.8760.43250.7750.26010.71180.70720.81590.52950.74720.85660.94240.36560.83330.66920.8737
16.14938.072817.31180.57070.75570.62980.40360.36840.64640.41750.78310.82750.450.68150.88220.49540.79230.32830.75880.70960.81590.54740.75280.87170.94360.33810.84440.70450.8847
15.38799.081916.72160.59070.76340.66110.350.38240.66360.41450.78080.82580.450.66630.88210.53050.80190.37480.75290.73370.81360.56580.75740.87380.93770.34210.83330.71440.8835
14.461410.091016.25660.62290.81720.68240.350.4010.71460.42790.78860.8310.50.71080.88020.540.81350.4050.71180.73720.81590.57670.76480.87830.94160.50190.88330.72090.8863

Framework versions

  • Transformers 4.50.0.dev0
  • Pytorch 2.6.0+cu124
  • Datasets 3.3.2
  • Tokenizers 0.21.0