Skip to content

If the training set does not contain a frame where all joints are labeled, crashes with IndexError #20

@cellistigs

Description

@cellistigs

Hello!

This issue is potentially related to #9 when training with a small number of frames. If there is not a single training frame in your training set with all joints labeled, the code will throw the error attached in the screenshot below:
Screen Shot 2021-08-26 at 10 50 52 AM

This is because all of the frames in my training set have at least one point missing. I have tracked this down to the method _compute_targets in the deepgraphpose.dataset module, line 599 (computing the max on joints across datapoints). Is there a reason not to take the number of joints directly from the config file? If not, I can make the change and submit a pull request.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions