diff --git a/src/main/java/sc/fiji/labkit/pixel_classification/classification/Trainer.java b/src/main/java/sc/fiji/labkit/pixel_classification/classification/Trainer.java index fefaaa70..1b5b3f67 100644 --- a/src/main/java/sc/fiji/labkit/pixel_classification/classification/Trainer.java +++ b/src/main/java/sc/fiji/labkit/pixel_classification/classification/Trainer.java @@ -109,7 +109,7 @@ public void trainLabeledFeatures( if (label == null) continue; LabelRegion region = regions.getLabelRegion(label); - Cursor cursor = region.cursor(); + Cursor cursor = region.inside().cursor(); while (cursor.hasNext()) { cursor.next(); ra.setPosition(cursor);