fix single index cutouts (#108)

jhuapl-boss · Jun 23, 2023 · b90e209 · b90e209
1 parent 8091885
commit b90e209
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/intern/service/cv/volume.py b/intern/service/cv/volume.py
@@ -66,8 +66,9 @@ def get_cutout(self, resource, res, x_range, y_range, z_range):
             x_range[0] : x_range[1], y_range[0] : y_range[1], z_range[0] : z_range[1]
         ]
 
-        # Remove channel dimension if num_channel = 1
-        data = np.squeeze(data)
+        # Remove channel dimension
+        if data.ndim == 4:
+            data = data[:,:,:,0]
         return data
 
     def delete_data(self, resource, res, x_range, y_range, z_range):