Affine

dataphy.visionpack.tforms.affine

Classes

RandomAffine(shear_deg: float = 5.0, perspective: float = 0.05, **kwargs: Any)

Bases: BaseTransform

Source code in src/dataphy/visionpack/tforms/affine.py

def __init__(
    self,
    shear_deg: float = 5.0,
    perspective: float = 0.05,
    **kwargs: Any
):
    super().__init__(**kwargs)
    self.shear_deg = shear_deg
    self.perspective = perspective

Attributes

shear_deg = shear_deg instance-attribute

perspective = perspective instance-attribute

p = p instance-attribute

apply_to = apply_to if apply_to is not None else ['rgb'] instance-attribute

sync_views = sync_views instance-attribute

update_intrinsics = update_intrinsics instance-attribute

mask_protect = mask_protect if mask_protect is not None else [] instance-attribute

min_visible_mask_pct = min_visible_mask_pct instance-attribute

resample = resample instance-attribute

border_mode = border_mode instance-attribute

pad_mode = pad_mode instance-attribute

pad_value = pad_value instance-attribute

seed_policy = seed_policy instance-attribute

Functions

forward(batch: Dict[str, Any]) -> Dict[str, Any]

Base forward method that handles cross-cutting logic.

Source code in src/dataphy/visionpack/tforms/base.py

def forward(self, batch: Dict[str, Any]) -> Dict[str, Any]:
    """Base forward method that handles cross-cutting logic."""
    if not self._should_apply():
        return batch

    # Setup seed policy
    episode_id = batch.get("episode_id")
    self._setup_seed(episode_id)

    # Apply transform
    return self._apply_transform(batch)