test implement of dlc-live

Author: JensBlack

DeepLabStream.py

@@ -264,6 +264,8 @@ def write_video(self, frames: dict, index: int):
     ######################
     @staticmethod
     def get_pose_mp(input_q, output_q):
+        from dlclive import DLCLive
+
         """
         Process to be used for each camera/DLC stream of analysis
         Designed to be run in an infinite loop
@@ -278,6 +280,19 @@ def get_pose_mp(input_q, output_q):
                 peaks = find_local_peaks_new(scmap, locref, ANIMALS_NUMBER, config)
                 output_q.put((index, peaks))
 
+
+        dlc_live = DLCLive(DLC_LIVE)
+        while True:
+            if input_q.full():
+                index, frame = input_q.get()
+                if not dlc_live.is_initialized:
+                    peaks = dlc_live.init_inference(frame)
+                else:
+                    peaks = dlc_live.get_pose(frame)
+
+                output_q.put((index, peaks))
+
+
     @staticmethod
     def create_mp_tools(devices):
         """

utils/poser.py

@@ -187,3 +187,31 @@ def create_animal_skeleton(dots_cluster: tuple) -> dict:
         animal_skeletons.append(create_animal_skeleton(unique_cluster))
 
     return animal_skeletons
+
+
+def transform_2skeleton(pose):
+    from utils.configloader import ALL_BODYPARTS
+    skeleton = dict()
+    counter = 0
+    for bp in pose:
+        skeleton[ALL_BODYPARTS[counter]] = tuple(np.array(bp[0:2],dtype = int))
+        counter += 1
+    return skeleton
+
+
+def transform_2pose(skeleton):
+    pose = np.array([*skeleton.values()])
+    return pose
+
+
+
+def calculate_skeletons_dlc_live(pose ,animals_number: int = 1) -> list:
+    """
+    Creating skeletons from given pose
+    There could be no more skeletons than animals_number
+    Only unique skeletons output
+    """
+
+    skeletons = [transform_2skeleton(pose)]
+
+    return skeletons