poseutils.datasets.unprocessed package

Submodules

poseutils.datasets.unprocessed.Dataset module

class poseutils.datasets.unprocessed.Dataset.Dataset(dataset_name)

Bases: object

Unprocessed dataset base class. Contains common methods shared across all dataset classes.

get_2d_train(jnts=14)

Returns 2d train data.

Parameters

jnts (int, optional) – Joint configuration (14 or 16), defaults to 14

Returns

Selected and sorted joint positions (NxMx2), M = 14 or 16

Return type

numpy.ndarray

get_2d_valid(jnts=14)

Returns 2d valid data.

Parameters

jnts (int, optional) – Joint configuration (14 or 16), defaults to 14

Returns

Selected and sorted joint positions (NxMx2), M = 14 or 16

Return type

numpy.ndarray

get_3d_train(jnts=14)

Returns 3d train data.

Parameters

jnts (int, optional) – Joint configuration (14 or 16), defaults to 14

Returns

Selected and sorted joint positions (NxMx3), M = 14 or 16

Return type

numpy.ndarray

get_3d_valid(jnts=14)

Returns 3d valid data.

Parameters

jnts (int, optional) – Joint configuration (14 or 16), defaults to 14

Returns

Selected and sorted joint positions (NxMx3), M = 14 or 16

Return type

numpy.ndarray

poseutils.datasets.unprocessed.GPADataset module

class poseutils.datasets.unprocessed.GPADataset.GPADataset(path)

Bases: poseutils.datasets.unprocessed.Dataset.Dataset

Dataset class for handling GPA dataset

Parameters

path (str) – path to npz file

load_data(path)

poseutils.datasets.unprocessed.H36MDataset module

class poseutils.datasets.unprocessed.H36MDataset.H36MDataset(path)

Bases: poseutils.datasets.unprocessed.Dataset.Dataset

Dataset class for H36M dataset.

Parameters

path (str) – Base directory path to h36m files with cameras.h5

load_3d_data(path, subjects, actions)

load_data(path)

project_to_cameras(poses_set)

poseutils.datasets.unprocessed.MPI3DHPDataset module

class poseutils.datasets.unprocessed.MPI3DHPDataset.MPI3DHPDataset(path)

Bases: poseutils.datasets.unprocessed.Dataset.Dataset

Dataset class for 3DHP dataset.

Parameters

path (str) – Path to compiled npz files for 3dhp dataset

load_data(path)

split_train_test(data)

poseutils.datasets.unprocessed.SURREALDataset module

class poseutils.datasets.unprocessed.SURREALDataset.SURREALDataset(path_train, path_valid)

Bases: poseutils.datasets.unprocessed.Dataset.Dataset

Dataset class for SURREAL dataset.

Parameters

• path_train (str) – Path to surreal training set npz

• path_valid (str) – Path to validation set npz

load_data(path_train, path_valid)

poseutils.datasets.unprocessed.TDPWDataset module

class poseutils.datasets.unprocessed.TDPWDataset.TDPWDataset(path)

Bases: poseutils.datasets.unprocessed.Dataset.Dataset

Dataset class for handling 3DPW dataset

Parameters

path (str) – path to npz file

load_data(path)

Module contents