solve bug for rdg color images

segment-anything-models-java · Nov 22, 2024 · e154ed3 · e154ed3
1 parent 35ba0d0
commit e154ed3
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/main/java/ai/nets/samj/ij/ui/IJComboBoxItem.java b/src/main/java/ai/nets/samj/ij/ui/IJComboBoxItem.java
@@ -22,6 +22,7 @@
 
 import ai.nets.samj.gui.components.ComboBoxItem;
 import ij.ImagePlus;
+import ij.plugin.CompositeConverter;
 import net.imglib2.RandomAccessibleInterval;
 import net.imglib2.img.Img;
 import net.imglib2.img.display.imagej.ImageJFunctions;
@@ -72,8 +73,10 @@ public String getImageName() {
 	 * Convert the {@link ImagePlus} into a {@link RandomAccessibleInterval}
 	 */
 	public <T extends RealType<T> & NativeType<T>> RandomAccessibleInterval<T> getImageAsImgLib2() {
-		Img<?> img = ImageJFunctions.wrap((ImagePlus) this.getValue());
-		return Cast.unchecked(img);
+		ImagePlus imp = (ImagePlus) this.getValue();
+		boolean isColorRGB = imp.getType() == ImagePlus.COLOR_RGB;
+		Img<?> image = ImageJFunctions.wrap(isColorRGB ? CompositeConverter.makeComposite(imp) : imp);
+		return Cast.unchecked(image);
 	}