Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

modified videodataset for window multiprocessing #8323

Open
wants to merge 6 commits into
base: dev
Choose a base branch
from
Open

modified videodataset for window multiprocessing #8323

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
068f3be
modified videodataset for window multiprocessing
JK-rez Feb 2, 2025
295d477
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Feb 2, 2025
30819b8
DCO Remediation Commit for JK-rez <[email protected]>
JK-rez Feb 2, 2025
62bc90b
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Feb 2, 2025
c085c4d
Merge branch 'dev' into videodataset
ericspod Feb 12, 2025
89e7c06
Merge branch 'dev' into videodataset
ericspod Mar 4, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 7 additions & 1 deletion monai/data/video_dataset.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,13 @@ def get_fps(self) -> int:
def get_frame(self) -> Any:
"""Get next frame. For a file, this will be the next frame, whereas for a camera
source, it will be the next available frame."""
ret, frame = self._get_cap().read()
#Assign explicit videocapture object to cap to realease file to avoid error in multiprocessing
if self.multiprocessing:
cap = self._get_cap()
ret, frame = cap.read()
cap.release()
else:
ret, frame = self._get_cap().read()
if not ret:
raise RuntimeError("Failed to read frame.")
# Switch color order if desired
Expand Down
Loading