Skip to content

bizy-coder/ViT_Geolocation

Repository files navigation

This is a collection of models and training for taking an image and predicting what US state it was from. FastAI was used for most model development. The training logs can be found at https://wandb.ai/ben_z/geolocation/?nw=nwuserben_z:

image

The accuracy is quite high. The initial model achieved 60% accuracy, with significantly better top-k accuracy. Later models pushed this up to 67%:

image

Finally, for those interested, here is a chart of what states were easy and hard to identify: image

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published