The mesonet package provides functions to build, train, obtain predictions from and visualize MesoNet variants.
It has been divided into four main modules:
model- This module provides functions to build and obtain predictions from MesoNet variants. It also consists of helper functions such as obtaining a classification report, saving the model to a file/directory, saving model history, etc.data- This module provides functions to apply augmentations and load images from the dataset for training and testing.train- This module provides a single function to train a MesoNet variant.visualization- This module provides functions to visualize loss curves and the intermediate convolutional layers.
There is also a utils module for the purpose of storing any miscellaneous utilities required in the package. Currently, it only stores the input size of images as the constant IMG_WIDTH. The default value is 256. Therefore, by default, any model you train will use 256 x 256 images as its input. You can easily change this by modifying the constant.