updated metadata plotting when cropping

Author: JensBlack

DeepLabStream.py

@@ -426,7 +426,13 @@ def get_analysed_frames(self) -> tuple:
         if self._dlc_running:
             analysed_frames = {}
             analysis_time = None
-            frame_width, frame_height = RESOLUTION
+
+            if CROP:
+                frame_height = CROP_Y[1] - CROP_Y[0]
+                frame_width = CROP_X[1] - CROP_X[0]
+            else:
+                frame_width,frame_height = RESOLUTION
+
             for camera in self._multiprocessing:
                 if self._multiprocessing[camera]['output'].full():
                     if self._start_time is None:

utils/plotter.py

@@ -9,7 +9,6 @@
 import cv2
 import numpy as np
 
-
 def plot_dots(image, coordinates, color, cond=False):
     """
     Takes the image and positional arguments from pose to plot corresponding dot
@@ -55,10 +54,11 @@ def plot_metadata_frame(image, frame_width, frame_height, current_fps, current_e
     """
     res_image = image.copy()
     font = cv2.FONT_HERSHEY_PLAIN
+
     cv2.putText(res_image, 'Time: ' + str(round(current_elapsed_time, 2)),
                 (int(frame_width * 0.8), int(frame_height * 0.9)), font, 1, (255, 255, 0))
     cv2.putText(res_image, 'FPS: ' + str(round(current_fps, 1)),
-                (int(frame_width * 0.8), int(frame_height * 0.95)), font, 1, (255, 255, 0))
+                (int(frame_width * 0.8), int(frame_height * 0.94)), font, 1, (255, 255, 0))
     return res_image
 
 def plot_dlc_bodyparts(image, bodyparts):