Skip to content

Grayscale

dataphy.visionpack.tforms.grayscale

Classes

ToGrayscale(keep_channels: int = 3, **kwargs: Any)

Bases: BaseTransform

Source code in src/dataphy/visionpack/tforms/grayscale.py 
11
12
13
14
15
16
17
def __init__(
    self,
    keep_channels: int = 3,
    **kwargs: Any
):
    super().__init__(**kwargs)
    self.keep_channels = keep_channels
Attributes
keep_channels = keep_channels 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 
 94
 95
 96
 97
 98
 99
100
101
102
103
104
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)