dima806/yoga_pose_image_classification

By dima806 · ↓ 15 · ♥ 4 · Updated 2024-10-19

What's in this repo

This page lists the files in dima806/yoga_pose_image_classification, including model.onnx.

1 ONNX models (.onnx files)
25 files, including 10 .json, 4 .pt, 3 .safetensors, 3 .bin, 2 .pth, 1 .md
Last updated 2024-10-19

ONNX files

File	Size	Download
onnx/model.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

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

Model info

image-classification transformers

transformers onnx safetensors vit image-classification base_model:google/vit-base-patch16-224-in21k base_model:quantized:google/vit-base-patch16-224-in21k license:apache-2.0 endpoints_compatible region:us

↓ 15 ♥ 4

Created: 2023-11-18

Updated: 2024-10-19

View on Hugging Face

Files (25)

.gitattributes

README.md

checkpoint-290/config.json

checkpoint-290/model.safetensors

checkpoint-290/optimizer.pt

checkpoint-290/preprocessor_config.json

checkpoint-290/rng_state.pth

checkpoint-290/scheduler.pt

checkpoint-290/trainer_state.json

checkpoint-290/training_args.bin

checkpoint-3800/config.json

checkpoint-3800/model.safetensors

checkpoint-3800/optimizer.pt

checkpoint-3800/preprocessor_config.json

checkpoint-3800/rng_state.pth

checkpoint-3800/scheduler.pt

checkpoint-3800/trainer_state.json

checkpoint-3800/training_args.bin

config.json

model.safetensors

onnx/config.json

onnx/model.onnx ONNX

onnx/preprocessor_config.json

preprocessor_config.json

training_args.bin

Related models

More from dima806

No other models by this author.

More image-classification models

Browse all image-classification models