Scene Classification

Fine-tuned vision transformer for scene recognition

Finetuned a ViT pretrained on ImageNet for the Kaggle Intel Classification Challenge.

  • Fine-tuning a vision transformer with PyTorch
  • Computer vision workflow for scene recognition
  • Hugging Face Space

Tech: PyTorch, Computer Vision, Gradio