unagi

• Modules
  • unagi.binarize
    • main()
    • parse_args()
  • unagi.dataset
    • ImageProcessor
      • ImageProcessor.process_img()
      • ImageProcessor.save_imgs()
      • ImageProcessor.split_img_overlay()
    • main()
    • parse_args()
    • process_img_wrapper()
  • unagi.train
    • ParallelDataGenerator
      • ParallelDataGenerator.augmentate_batch()
      • ParallelDataGenerator.on_epoch_end()
    • main()
    • parse_args()
  • unagi.cli
    • get_version()
    • main()
• Sub Packages
  • unagi.utils
    • unagi.utils.augmentor_utils
      • GaussianNoiseAugmentor
        • GaussianNoiseAugmentor.perform_operation()
      • InvertPartAugmentor
        • InvertPartAugmentor.perform_operation()
      • SaltPepperNoiseAugmentor
        • SaltPepperNoiseAugmentor.perform_operation()
    • unagi.utils.callback_utils
      • Visualisation
        • Visualisation.on_epoch_end()
        • Visualisation.on_train_end()
      • create_callbacks()
    • unagi.utils.img_processing_utils
      • ImageUtils
        • ImageUtils.add_border()
        • ImageUtils.binarize_img()
        • ImageUtils.combine_imgs()
        • ImageUtils.mkdir_s()
        • ImageUtils.normalize_gt()
        • ImageUtils.normalize_in()
        • ImageUtils.postprocess_img()
        • ImageUtils.process_with_unagi()
        • ImageUtils.shuffle_imgs()
        • ImageUtils.split_img()
    • unagi.utils.metric_utils
      • ModelMetrics
        • ModelMetrics.dice_coef()
        • ModelMetrics.dice_coef_loss()
        • ModelMetrics.focal_loss()
        • ModelMetrics.focal_tversky()
        • ModelMetrics.jacard_coef()
        • ModelMetrics.jacard_coef_loss()
        • ModelMetrics.tversky()
        • ModelMetrics.tversky_loss()
        • ModelMetrics.weighted_cross_entropy()
    • unagi.utils.model_utils
      • ModelLayers
        • ModelLayers.double_conv_layer()
        • ModelLayers.down_layer()
        • ModelLayers.up_layer()
      • UNAGIModel
        • UNAGIModel.unet()