README.md

GUI-Based-Training

A streamlit web app that allows you to train Few Shot image classification models using Prototypical Networks with Resnet-18 as the backbone network. During Training, model parameters are also saved on disk.

Preparing and Loading The Data

Model Training and Saving