diplomat.tweak

diplomat.tweak(videos: List[PathLike | str] | PathLike | str, csvs: List[PathLike | str] | PathLike | str, **kwargs) → None

Make modifications to DIPLOMAT produced tracking results created for a video using a limited version of the interactive labeling UI. Allows for touching up and fixing any minor issues that may arise after tracking and saving results.

Parameters:

• videos – Paths to video file(s) corresponding to the provided csv files.

• csvs – The path (or list of paths) to the csv file(s) to edit.

• kwargs –

  Additional visual arguments:

  • pcutoff (Type: RangedFloat[min=0.0, max=1.0], Default: 0.1): The probability to cutoff results below.

  • dotsize (Type: int, Default: 4): The size of the dots.

  • alphavalue (Type: RangedFloat[min=0.0, max=1.0], Default: 0.7): The alpha value of the dots.

  • colormap (Type: to_colormap, Default: None): The colormap to use for tracked points in the video. Can be a matplotlib colormap or a list of matplotlib colors.

  • shape_list (Type: Optional[List[str]], Default: None): A list of shape names, shapes to use for drawing each individual’s dots.

  • line_thickness (Type: int, Default: 1): Thickness of lines drawn.

  • skeleton (Type: Union[List[Tuple[str, str]], Dict[str, List[str]], List[str], None, bool], Default: None): The skeleton to use for this this run of DIPLOMAT. Defaults to None, which uses the skeleton associated with the project. Can be a list of strings, a list of tuples of strings, a dictionary of strings to strings or lists of strings, or True/False (True connects all parts, False connects no parts, disabling the skeleton).