fix bug in to_ndarray causing memory problems

pattonw · pattonw · commit f4237490cd40 · 2024-10-15T09:45:15.000-07:00
full array was being requested in memory to check shape
diff --git a/funlib/persistence/arrays/array.py b/funlib/persistence/arrays/array.py
@@ -339,7 +339,7 @@ def to_ndarray(self, roi, fill_value=0):
 
         shape = roi.shape / self.voxel_size
         data = np.zeros(
-            self[self.roi].shape[: self.channel_dims] + shape, dtype=self.data.dtype
+            self.shape[: self.channel_dims] + shape, dtype=self.data.dtype
         )
         if fill_value != 0:
             data[:] = fill_value

Original file line number	Diff line number	Diff line change
`@@ -339,7 +339,7 @@ def to_ndarray(self, roi, fill_value=0):`
`339`	`339`
`340`	`340`	`shape = roi.shape / self.voxel_size`
`341`	`341`	`data = np.zeros(`
`342`		`- self[self.roi].shape[: self.channel_dims] + shape, dtype=self.data.dtype`
	`342`	`+ self.shape[: self.channel_dims] + shape, dtype=self.data.dtype`
`343`	`343`	`)`
`344`	`344`	`if fill_value != 0:`
`345`	`345`	`data[:] = fill_value`