dima806
/

weather_types_image_detection

Image Classification

Inference Endpoints

Model card Files Files and versions Community

dima806 commited on 30 days ago

Commit

d6cad40

•

1 Parent(s): 9f930cc

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -3,8 +3,10 @@ license: apache-2.0
 metrics:
 - accuracy
 - f1
 ---
-Returns weather type given an image.
 See https://www.kaggle.com/code/dima806/weather-types-image-prediction-vit for more details.

 metrics:
 - accuracy
 - f1
+base_model:
+- google/vit-base-patch16-224-in21k
 ---
+Returns weather type given an image with about 96% accuracy.
 See https://www.kaggle.com/code/dima806/weather-types-image-prediction-vit for more details.