augmentations

xvr.model.augmentations

RandomCenterCrop

RandomCenterCrop(maxcrop: int, p: float = 0.5)

Simulate collimation.

Source code in src/xvr/model/augmentations.py

def __init__(self, maxcrop: int, p: float = 0.5):
    super().__init__(p=p)
    self.maxcrop = maxcrop