diplomat.frontends.sleap.tweak_results_sleap.tweak_videos

diplomat.frontends.sleap.tweak_results_sleap.tweak_videos(config: Union[PathLike, str], videos: Union[List[Union[PathLike, str]], Union[PathLike, str]], **kwargs) → None[source]

Make minor modifications and tweaks to sleap results (stored in .slp files) using DIPLOMAT’s supervised UI.

Parameters:

  • config – The path (or list of paths) to the sleap model(s) used for inference, each as either as a folder or zip file.

  • videos – Paths to the sleap label files, or .slp files, to make minor modifications to, NOT the video files.

  • kwargs –

    The following additional arguments are supported:

    • 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 sleap project. Can be a list of strings, a list of tuples of strings, a dictionary of strings to strings or listsof strings, or True/False (True connects all parts, False connects no parts, disabling the skeleton).