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.
harshvirsandhu / gui-based-training Goto Github PK
View Code? Open in Web Editor NEWA streamlit web app that allows you to train Few Shot image classification models