Don't assume YXCZ stack when creating accessor

model_server/extensions/ilastik/models.py

@@ -262,7 +262,7 @@ class IlastikObjectClassifierFromPixelPredictionsModel(IlastikModel, ImageToImag
             raise InvalidInputImageError('Expecting input image and pixel probabilities to be the same shape')
         pxch = kwargs.get('pixel_classification_channel', 0)
         pxtr = kwargs.get('pixel_classification_threshold', 0.5)
-        mask = InMemoryDataAccessor(pxmap.get_one_channel_data(pxch).data > pxtr)
+        mask = img._derived_accessor(pxmap.get_one_channel_data(pxch).data > pxtr)
         obmap, _ = self.infer(img, mask)
         return obmap