Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion CITATION.cff
Original file line number Diff line number Diff line change
Expand Up @@ -52,4 +52,4 @@ authors:
date-released: 2020-08-05
doi: "10.7554/eLife.61909"
license: "AGPL-3.0-or-later"
version: "1.0.3"
version: "1.1.0"
38 changes: 22 additions & 16 deletions dlclive/benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,14 @@
"""

import csv
import os
import platform
import subprocess
import sys
import time
import warnings
from pathlib import Path
from typing import List, Optional, Tuple, Union

import colorcet as cc
import cv2
Expand Down Expand Up @@ -65,23 +67,27 @@ def show_progress(count, block_size, total_size):


def benchmark_videos(
model_path,
model_type,
video_path,
output=None,
n_frames=1000,
model_path: str,
model_type: str,
video_path: Union[str, List[str]],
output: Optional[str] = None,
n_frames: int = 1000,
tf_config=None,
resize=None,
pixels=None,
cropping=None,
dynamic=(False, 0.5, 10),
print_rate=False,
display=False,
pcutoff=0.5,
display_radius=3,
cmap="bmy",
save_poses=False,
save_video=False,
device: str | None = None,
resize: Optional[Union[float, List[float]]] = None,
pixels: Optional[Union[int, List[int]]] = None,
cropping: Optional[List[int]] = None,
dynamic: Tuple[bool, float, int] = (False, 0.5, 10),
print_rate: bool = False,
precision: str = "FP32",
display: bool = False,
pcutoff: float = 0.5,
display_radius: int = 3,
cmap: str = "bmy",
save_poses: bool = False,
save_video: bool = False,
single_animal: bool = True,
draw_keypoint_names: bool = False,
):
"""Analyze videos using DeepLabCut-live exported models.
Analyze multiple videos and/or multiple options for the size of the video
Expand Down
2 changes: 1 addition & 1 deletion dlclive/version.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,5 +6,5 @@
Licensed under GNU Lesser General Public License v3.0
"""

__version__ = "3.0.0a0"
__version__ = "1.1.0"
VERSION = __version__
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "deeplabcut-live"
version = "3.0.0a0"
version = "1.1.0"
description = "Class to load exported DeepLabCut networks and perform pose estimation on single frames (from a camera feed)"
authors = ["A. & M. Mathis Labs <admin@deeplabcut.org>"]
license = "AGPL-3.0-or-later"
Expand Down
Loading