From 03e39960e31bb358ba3f112220a5fdbb2e6acc76 Mon Sep 17 00:00:00 2001 From: Chris Broz Date: Mon, 7 Oct 2024 09:08:28 -0500 Subject: [PATCH] DLC `make_video` minor fixes (#1150) * Pass output file name as str * Update changelog --- CHANGELOG.md | 1 + src/spyglass/position/v1/dlc_utils_makevid.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 9151ebb3c..63aef8498 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -53,6 +53,7 @@ dj.FreeTable(dj.conn(), "common_session.session_group").drop() - Fix video directory bug in `DLCPoseEstimationSelection` #1103 - Restore #973, allow DLC without position tracking #1100 - Minor fix to `DLCCentroid` make function order #1112, #1148 + - Pass output path as string to `cv2.VideoWriter` #1150 - Spike Sorting diff --git a/src/spyglass/position/v1/dlc_utils_makevid.py b/src/spyglass/position/v1/dlc_utils_makevid.py index 5ee7c096a..994c027f0 100644 --- a/src/spyglass/position/v1/dlc_utils_makevid.py +++ b/src/spyglass/position/v1/dlc_utils_makevid.py @@ -134,7 +134,7 @@ def _init_video(self): def _init_cv_video(self): _ = self._init_video() self.out = cv2.VideoWriter( - filename=self.output_video_filename, + filename=str(self.output_video_filename), fourcc=cv2.VideoWriter_fourcc(*"mp4v"), fps=self.frame_rate, frameSize=self.frame_size,