dgs.models.submission.posetrack21.PoseTrack21Submission

class dgs.models.submission.posetrack21.PoseTrack21Submission(config: dict[str, any], path: list[str])[source]

Class for creating and appending to a PoseTrack21 -style submission file.

__init__(config: dict[str, any], path: list[str]) None[source]

Methods

append(s, *_args, **_kwargs)

Given data, append to the created PoseTrack21 submission file.

clear()

Clear the submission data.

configure_torch_module(module[, train])

Set compute mode and send model to the device or multiple parallel devices if applicable.

get_anno_data(s)

Given a State, extract data for the 'annotations' list used in the submission file.

get_image_data(s)

Given a State, extract data for the 'images' used in the submission file.

save()

Save the submission data in a file.

terminate()

Terminate the submission file creation.

validate_params(validations[, attrib_name])

Given per key validations, validate this module's parameters.

Attributes

device

Get the device of this module.

is_training

Get whether this module is set to training-mode.

module_name

Get the name of the module.

module_type

name

Get the name of the module.

name_safe

Get the escaped name of the module usable in filepaths by replacing spaces and underscores.

precision

Get the (floating point) precision used in multiple parts of this module.

data

fp

format