BaseImageTestTimeAugmentation

class easypl.callbacks.predictors.base_image.BaseImageTestTimeAugmentation(n: int, augmentations: List, augmentation_method: str = 'first', phase='val')

Abstract image base callback for test-time-augmentation.

n

Number of augmentations.

Type

int

augmentations

List of augmentations, which will be used.

Type

List

augmentation_method

Method of selecting augmentations from list. Available: [“first”, “random”]

Type

str

phase

Phase which will be used by this predictor callback. Available: [“val”, “test”, “predict”].

Type

str

post_init(trainer: Trainer, pl_module: LightningModule)

Method for initialization in first batch handling. [NOT REQUIRED]

trainer

Trainer of pytorch-lightning

Type

pytorch_lightning.Trainer

pl_module

LightningModule of pytorch-lightning

Type

pytorch_lightning.LightningModule