CoolFace
Modelpublic

ernie-ai/autotrain-document-text-language-ar-en-zh-3338392240

sourceHugging Faceupdated 4y agoView on Hugging Face
1likes16downloads
Model Card

finetuned-MS-swin-doc-text-classifer

This model is a fine-tuned version of Microsoft’s Swin Transformer tiny-sized model microsoft/swin-tiny-patch4-window7-224 on the ernie-ai/image-text-examples-ar-cn-latin-notext dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.267
  • —Accuracy: 0.882

Model description

It is an image classificatin model fine-tuned to predict whether an images contains text and if that text is Latin script, Chinese or Arabic. It also classifies non-text images.

Training and evaluation data

Dataset: [ernie-ai/image-text-examples-ar-cn-latin-notext]

Model Trained Using AutoTrain

  • —Problem type: Multi-class Classification
  • —Model ID: 3338392240
  • —CO2 Emissions (in grams): 2.2267

Validation Metrics

  • —Loss: 0.267
  • —Accuracy: 0.882
  • —Macro F1: 0.862
  • —Micro F1: 0.882
  • —Weighted F1: 0.880
  • —Macro Precision: 0.877
  • —Micro Precision: 0.882
  • —Weighted Precision: 0.883
  • —Macro Recall: 0.856
  • —Micro Recall: 0.882
  • —Weighted Recall: 0.882