dgs.models.embedding_generator.pose_based

Different pose based embedding generators.

Module Classes

KeyPointConvolutionPBEG(*args, **kwargs)

Create a short torch Module that has one convolutional layer reducing the key points using relational information and an arbitrary number of hidden fully connected layers at the end.

LinearPBEG(*args, **kwargs)

Model to compute a pose-embedding given a pose, or batch of poses describing them as a single vector.