Skip to main content Link Search Menu Expand Document (external link)

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.