deepof.visuals module

deepof.visuals

General plotting functions for the deepof package.

deepof.visuals.plot_arena(coordinates, ...)

Plot the arena in the given canvas.

deepof.visuals.heatmap(dframe, bodyparts[, ...])

Return a heatmap of the movement of a specific bodypart in the arena.

deepof.visuals.plot_heatmaps(coordinates, ...)

Plot heatmaps of the specified body parts (bodyparts) of the specified animal (i).

deepof.visuals.plot_gantt(coordinates, ...)

Return a scatter plot of the passed projection.

deepof.visuals.plot_enrichment(coordinates)

Violin plots per cluster per condition.

deepof.visuals.plot_transitions(coordinates, ...)

Compute and plots transition matrices for all data or per condition.

deepof.visuals.plot_stationary_entropy(...)

Compute and plots transition stationary distribution entropy per condition.

deepof.visuals.plot_normative_log_likelihood(...)

Plot a bar chart with normative log likelihoods per experimental condition, and compute statistics.

deepof.visuals.plot_embeddings(coordinates)

Return a scatter plot of the passed projection.

deepof.visuals.animate_skeleton(coordinates, ...)

Render a FuncAnimation object with embeddings and/or motion trajectories over time.

deepof.visuals.plot_cluster_detection_performance(...)

Plot either a confusion matrix or a bar chart with balanced accuracy for cluster detection cross validated models.

deepof.visuals.plot_shap_swarm_per_cluster(...)

Plot a swarm plot of the SHAP values for a given cluster.

deepof.visuals.output_cluster_video(cap, ...)

Output a video with the frames corresponding to the cluster.

deepof.visuals.output_videos_per_cluster(...)

Given a list of videos, and a list of soft counts per video, outputs a video for each cluster.

deepof.visuals.output_unsupervised_annotated_video(...)

Given a video, and soft_counts per frame, outputs a video with the frames annotated with the cluster they belong to.

deepof.visuals.export_annotated_video(...[, ...])

Export annotated videos from both supervised and unsupervised pipelines.

deepof.visuals.plot_distance_between_conditions(...)

Plot the distance between conditions across a growing time window.

deepof.visuals.tag_annotated_frames(frame, ...)

Annotate a given frame with on-screen information about the recognised patterns.

deepof.visuals.annotate_video(coordinates, ...)

Render a version of the input video with all supervised taggings in place.