pocketpose.models.decoder.base_decoder
#
Module Contents#
Classes#
Base class for all decoders. |
- class pocketpose.models.decoder.base_decoder.Decoder#
Bases:
abc.ABC
Base class for all decoders.
Decoders are used to decode the prediction of pose models into a keypoint list in the image coordinate system. The keypoint list is a list of tuples (x, y, score) where x and y are the coordinates and score is the prediction confidence.
All decoders must implement the decode method. Each model has a corresponding decoder, and the decode method is automatically called when the model is used for prediction.
- abstract decode(x, image_size)#