Merge pull request #148 from deruyter92/dlclive3

Resolve merge-conflicts in dlclive3 with main

Author: deruyter92

CITATION.cff

@@ -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"

dlclive/benchmark.py

@@ -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
@@ -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

dlclive/version.py

@@ -6,5 +6,5 @@
 Licensed under GNU Lesser General Public License v3.0
 """
 
-__version__ = "3.0.0a0"
+__version__ = "1.1.0"
 VERSION = __version__

pyproject.toml

@@ -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 <[email protected]>"]
 license = "AGPL-3.0-or-later"