Updated PySide2 import for consistency

removed old todo
removed debug print in generic.py

Author: JensBlack

DeepLabStream.py

@@ -284,8 +284,8 @@ def get_pose_mp(input_q, output_q):
                     start_time = time.time()
                     if MODEL_ORIGIN == 'DLC':
                         scmap, locref, pose = get_pose(frame, config, sess, inputs, outputs)
-                        # TODO: Remove alterations to original
                         peaks = find_local_peaks_new(scmap, locref, ANIMALS_NUMBER, config)
+                        #Use the line below to use raw DLC output rather then DLStream optimization
                         # peaks = pose
                     if MODEL_ORIGIN == 'MADLC':
                         peaks = get_ma_pose(frame, config, sess, inputs, outputs)

utils/generic.py

@@ -122,7 +122,6 @@ def enable_stream(self, resolution, framerate, *args):
         (hopefully)
         """
         # set video to first frame
-        print('Thinking of beginning things...')
         self._camera = cv2.VideoCapture(VIDEO_SOURCE)
         self._camera.set(cv2.CAP_PROP_POS_FRAMES,0)
 

utils/gui_image.py

@@ -9,9 +9,11 @@
 from utils.configloader import RESOLUTION
 
 import cv2
-from PyQt5.QtCore import QObject, pyqtSignal, Qt, pyqtSlot
-from PyQt5.QtWidgets import QWidget, QLabel
-from PyQt5.QtGui import QImage, QPixmap
+from PySide2.QtCore import QObject
+from PySide2.QtCore import Signal as pyqtSignal
+from PySide2.QtCore import Slot as pyqtSlot
+from PySide2.QtWidgets import QWidget, QLabel
+from PySide2.QtGui import QImage, QPixmap
 
 width, height = RESOLUTION