Edit model card

Returns yoga pose given an image.

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
Downloads last month
30
Safetensors
Model size
85.8M params
Tensor type
F32
·

Spaces using dima806/yoga_pose_image_classification 2