s6.nn.data package

Submodules

s6.nn.data.augmentation module

class s6.nn.data.augmentation.RawAugmentation(gamma_range=(0.6, 1.4), brightness_shift_range=(-0.2, 0.2), contrast_range=(0.7, 1.4))

Bases: object

s6.nn.data.dataset module

class s6.nn.data.dataset.KeypointDataset(dataset_dirs: List[str], target_names: List[str], output_size: Tuple[int, int] = (224, 224), crop_range: Tuple[float, float] = (0.95, 1), transform: callable | None = None)

Bases: Dataset

center_crop(image: ndarray, keypoints: Tensor, crop_range=(0.5, 1.0), padding=0.15)

scale(image: any, keypoints: Tensor)

Module contents

Data loading and augmentation subpackage for S6 neural nets.