data_loader.py File Reference

Low-level dataset loading utilities for pose estimation. More...

Go to the source code of this file.

Namespaces
	data_loader

Functions
def	data_loader._parse_labels (str label_path)
def	data_loader._normalize_labels (label, int n, int l)
tf.data.Dataset	data_loader._get_path_dataset (str path, int seed, bool shuffle=True)
def	data_loader._get_padded_labels (data, r, R, height, width)
tuple[tf.Tensor, tf.Tensor]	data_loader._preprocess_function (tf.Tensor data_x, tf.Tensor data_y, tuple[int] image_size, str interpolation, str aspect_ratio, str color_mode, int nbr_keypoints)
Tuple[tf.data.Dataset, tf.data.Dataset]	data_loader._get_train_val_ds (str training_path, tuple[int] image_size=None, int nbr_keypoints=None, str interpolation=None, str aspect_ratio=None, str color_mode=None, float validation_split=None, int batch_size=None, int seed=None, bool shuffle=True, bool to_cache=False)
tf.data.Dataset	data_loader._get_ds (str data_path=None, tuple[int] image_size=None, int nbr_keypoints=None, str interpolation=None, str aspect_ratio=None, str color_mode=None, int batch_size=None, int seed=None, bool shuffle=False, bool to_cache=False)
Tuple[tf.data.Dataset, tf.data.Dataset, tf.data.Dataset]	data_loader.load_dataset (str dataset_name=None, str training_path=None, str validation_path=None, str quantization_path=None, str test_path=None, float validation_split=None, int nbr_keypoints=None, tuple[int] image_size=None, str interpolation=None, str aspect_ratio=None, str color_mode=None, int batch_size=None, int seed=None)

Detailed Description

Low-level dataset loading utilities for pose estimation.

Handles COCO JSON annotation parsing, image loading, and keypoint coordinate normalization. Used by preprocess.py.

Note

Politecnico di Milano, A.Y. 2024-2025. Multidisciplinary Project — Neural Network Deployment on STM32N6 NPU.

Definition in file data_loader.py.