dima806
/

wildfire_types_image_detection

Image Classification

Model card Files Files and versions

dima806 commited on Oct 19, 2024

Commit

7c122d2

·

verified ·

1 Parent(s): 05cae24

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 wildfire type given an image.
 See https://www.kaggle.com/code/dima806/wildfire-image-detection-vit for more details.

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