Skip to content

Wb Shift

dataphy.visionpack.tforms.wb_shift

Classes

WBShift(kelvin_shift: Optional[float] = None, mired_shift: Optional[float] = None, per_channel_gain: List[float] = [0.95, 1.05], preserve_robot_color: bool = True, **kwargs: Any)

Bases: BaseTransform

Source code in src/dataphy/visionpack/tforms/wb_shift.py 
11
12
13
14
15
16
17
18
19
20
21
22
23
def __init__(
    self,
    kelvin_shift: Optional[float] = None,
    mired_shift: Optional[float] = None,
    per_channel_gain: List[float] = [0.95, 1.05],
    preserve_robot_color: bool = True,
    **kwargs: Any
):
    super().__init__(**kwargs)
    self.kelvin_shift = kelvin_shift or 1000.0
    self.mired_shift = mired_shift or 10.0
    self.per_channel_gain = per_channel_gain
    self.preserve_robot_color = preserve_robot_color
Attributes
kelvin_shift = kelvin_shift or 1000.0 instance-attribute
mired_shift = mired_shift or 10.0 instance-attribute
per_channel_gain = per_channel_gain instance-attribute
preserve_robot_color = preserve_robot_color 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)