U-Net

• dataset module
  • SegmentationDataset
    • SegmentationDataset.image_dir
    • SegmentationDataset.mask_dir
    • SegmentationDataset.transform
    • SegmentationDataset.images
    • SegmentationDataset.__len__()
    • SegmentationDataset.__getitem__()
• trainer module
  • Trainer
    • Trainer.model
    • Trainer.optimizer
    • Trainer.train_loader
    • Trainer.val_loader
    • Trainer.model_is_binary
    • Trainer.train_loss
    • Trainer.val_loss
    • Trainer.val_accuracy
    • Trainer.val_dice
    • Trainer.loss_fn
    • Trainer.device
    • Trainer.scaler
    • Trainer.train_step()
    • Trainer.val_step()
    • Trainer.save_checkpoint()
    • Trainer.load_checkpoint()
    • Trainer.train()
    • Trainer.load_checkpoint()
    • Trainer.save_checkpoint()
    • Trainer.train()
    • Trainer.train_step()
    • Trainer.val_step()
• unet module
  • DoubleConvolution
    • DoubleConvolution.double_conv
    • DoubleConvolution.forward()
    • DoubleConvolution.forward()
  • UNet
    • UNet.num_features
    • UNet.ups
    • UNet.downs
    • UNet.pool
    • UNet.bottleneck
    • UNet.final_conv
    • UNet.is_binary()
    • UNet.forward()
    • UNet.forward()
    • UNet.is_binary()