Allow setting StarDist data 2d/3d

Author: Stefan Hahmann

src/main/java/org/mastodon/mamut/detection/StarDist.java

@@ -151,6 +151,11 @@ public ModelType getModelType()
 		return modelType;
 	}
 
+	public void setDataIs2D( final boolean dataIs2D )
+	{
+		this.dataIs2D = dataIs2D;
+	}
+
 	private String getImportStarDistCommand()
 	{
 		if ( modelType.getModelPath() == null )

src/main/java/org/mastodon/mamut/detection/StarDistDetector.java

@@ -135,6 +135,7 @@ public void compute( final List< SourceAndConverter< ? > > sources, final ModelG
 				 * will have a size of 1.
 				 */
 				Img< ? > segmentation = starDist.segmentImage( Cast.unchecked( image ) );
+				starDist.setDataIs2D( !isData3D );
 
 				final AffineTransform3D transform = DetectionUtil.getTransform( sources, timepoint, setup, level );
 				LabelImageUtils.createSpotsForFrame( graph, Cast.unchecked( segmentation ), timepoint, transform, 1d );