Computer Vision Experimentation Frameworks - Image Classification
Keywords: deep learning, classification, regression, out-of-the-box, pytorch-implementation
This solution contains a framework for training deep learning-based classification and regression models with Pytorch Lightning. It comprises several architectures, regularization, augmentation and training techniques and aims to provide an easy-to-use suite for experimenting and fast model development using your own datasets. Additionally, extensive experiments on the CIFAR10 dataset provide valuable insights into the different features of this framework and their impact on model performance.