Skip to content

Commit 5de4167

Browse files
committed
Add error for multi-channel images
1 parent cae87d6 commit 5de4167

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

src/torchio/visualization.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -301,6 +301,9 @@ def make_video(
301301
if duration is not None and frame_rate is not None:
302302
message = 'Provide either duration or frame_rate, not both.'
303303
raise ValueError(message)
304+
if len(tensor) > 1:
305+
message = 'Only single-channel tensors are supported for video output for now.'
306+
raise ValueError(message)
304307
frames = tensor.numpy()[0].T
305308
num_frames = len(frames)
306309
if duration is not None:

0 commit comments

Comments
 (0)