Add logo and fix ROI img reading

MouseLand · Feb 22, 2023 · c525841 · c525841
1 parent ffa9845
commit c525841
Show file tree

Hide file tree

Showing 6 changed files with 2 additions and 2 deletions.
diff --git a/facemap/gui/ops_user.npy b/facemap/gui/ops_user.npy
diff --git a/facemap/mouse.png b/facemap/mouse.png
diff --git a/facemap/pose/pose_gui.py b/facemap/pose/pose_gui.py
@@ -112,7 +112,7 @@ def __init__(self, frame, video_id, gui, pose, last_video):
             int(np.floor(window_max_size.height() * fraction * aspect_ratio)),
         )
         self.gui = gui
-        self.frame = frame
+        self.frame = frame.squeeze()
         self.pose = pose
         self.last_video = last_video
         self.setWindowTitle("Select face ROI for video: " + str(video_id))

diff --git a/facemap/pose/refine_pose.py b/facemap/pose/refine_pose.py
@@ -999,7 +999,7 @@ def get_frames_from_indices(self, indices):
         for i, index in enumerate(indices):
             frame = utils.get_frame(
                 index, self.gui.nframes, self.gui.cumframes, self.gui.video
-            )[0]
+            )[0].squeeze()
             # Convert to grayscale
             imall[i] = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
             # Convert to int

diff --git a/figs/mouse.png b/figs/mouse.png
diff --git a/figs/mouse_old.png b/figs/mouse_old.png