Skip to content

Commit fb22843

Browse files
small bug in list input
1 parent 0501b4e commit fb22843

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cellpose/denoise.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -588,7 +588,7 @@ def eval(self, x, batch_size=8, channels=None, channel_axis=None, z_axis=None,
588588
diameter = self.dn.diam_mean if self.dn.ratio > 1 else diameter
589589
masks, flows, styles = self.cp.eval(
590590
img_restore, batch_size=batch_size, channels=channels_new, channel_axis=-1,
591-
z_axis=0 if img_restore.ndim > 3 and img_restore.shape[0] > 0 else None,
591+
z_axis=0 if not isinstance(img_restore, list) and img_restore.ndim > 3 and img_restore.shape[0] > 0 else None,
592592
normalize=normalize_params, rescale=rescale, diameter=diameter,
593593
tile_overlap=tile_overlap, augment=augment, resample=resample,
594594
invert=invert, flow_threshold=flow_threshold,

0 commit comments

Comments
 (0)