s6.nn.data.dataset¶
- 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)