PatchStackObjectClassifier now returns a valid MonoPatchStack

extensions/chaeo/models.py

@@ -33,11 +33,12 @@ class PatchStackObjectClassifier(IlastikObjectClassifierFromSegmentationModel):
 
         # for some reason ilastik scrambles these axes to Z(1)YX(1)
         assert obmaps[0].shape == (input_acc.nz, 1, input_acc.hw[0], input_acc.hw[1], 1)
-        yxcz = np.moveaxis(
-            obmaps[0][:, :, :, :, 0],
-            [2, 3, 1, 0],
-            [0, 1, 2, 3]
+        yxz = np.moveaxis(
+            obmaps[0][:, 0, :, :, 0],
+            [1, 2, 0],
+            [0, 1, 2]
         )
 
-        assert yxcz.shape == input_acc.shape
-        return InMemoryDataAccessor(data=yxcz), {'success': True}
\ No newline at end of file
+        assert yxz.shape[0:2] == input_acc.hw
+        assert yxz.shape[2] == input_acc.nz
+        return MonoPatchStack(data=yxz), {'success': True}
\ No newline at end of file