pipert.contrib.pose_estimation

Module Contents

Classes

DefaultPredictor	Create a simple end-to-end predictor with the given config.
PoseEstLogic	Helper class that provides a standard way to create an ABC using
PoseEstComponent

class pipert.contrib.pose_estimation.DefaultPredictor(cfg)

Create a simple end-to-end predictor with the given config. The predictor takes an BGR image, resizes it to the specified resolution, runs the model and produces a dict of predictions.

Attributes:

metadata (Metadata): the metadata of the underlying dataset, obtained from

cfg.DATASETS.TEST.

__call__(self, original_image)

Args:

original_image (np.ndarray): an image of shape (H, W, C) (in BGR order).

Returns:

predictions (dict): the output of the model

class pipert.contrib.pose_estimation.PoseEstLogic(in_queue, out_queue, *args, **kwargs)

Bases: pipert.core.routine.Routine

Helper class that provides a standard way to create an ABC using inheritance.

main_logic(self, *args, **kwargs)

setup(self, *args, **kwargs)

cleanup(self, *args, **kwargs)

class pipert.contrib.pose_estimation.PoseEstComponent(in_key, out_key, redis_url, maxlen=100, endpoint='')

Bases: pipert.core.component.BaseComponent

pipert.contrib.pose_estimation.parser