Sub Packages

unagi.utils

Unagi has one sub package called utils which contains several methods and classes that are used to create the dataset and train the model.

unagi.utils.augmentor_utils

Different augmentation objects are created to use during the training. This module contains those different augmentation objects used in the training.

unagi.utils.callback_utils

Callbacks for visulations of the training and saving the model are created using the objects in callback_utils module.

unagi.utils.img_processing_utils

Different image manipulation techniques are used in UNAGI. Dataset module uses different methods to create the images parts and preprocess the images during training.

unagi.utils.metric_utils

UNAGI uses different loss functions and metrics to measure the performace of the model training. This module contains different loss functions and metrics used in the training and validation of the model.

unagi.utils.model_utils

Contains methods used to create the U-net architechture used in the model training such as DoubelConv layer, upsampling, and downsampling paths.