dima806's picture
Update README.md
b80c749 verified
metadata
license: apache-2.0
metrics:
  - accuracy
  - f1
base_model:
  - google/vit-base-patch16-224-in21k

Returns yoga pose given an image with 97% accuracy.

See https://www.kaggle.com/code/dima806/yoga-pose-classification-vit for more details.

Classification report:

                   precision    recall  f1-score   support

           Bridge     0.9776    0.9776    0.9776       134
            Child     0.9701    0.9701    0.9701       134
            Cobra     0.9774    0.9701    0.9738       134
     Downward-Dog     0.9924    0.9701    0.9811       134
           Pigeon     1.0000    0.9851    0.9925       134
Standing-Mountain     0.9214    0.9699    0.9451       133
             Tree     0.9767    0.9474    0.9618       133
         Triangle     0.9270    0.9549    0.9407       133
          Warrior     0.9699    0.9627    0.9663       134

         accuracy                         0.9676      1203
        macro avg     0.9681    0.9676    0.9677      1203
     weighted avg     0.9681    0.9676    0.9677      1203