avoid permutation of axes

segment-anything-models-java · Apr 24, 2024 · 2deed03 · 2deed03
1 parent 92e622b
commit 2deed03
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/main/java/ai/nets/samj/ij/ui/IJ1PromptsProvider.java b/src/main/java/ai/nets/samj/ij/ui/IJ1PromptsProvider.java
@@ -195,13 +195,10 @@ private RoiManager startRoiManager() {
 	 */
 	public RandomAccessibleInterval<?> giveProcessedSubImage(SAMModel selectedModel) {
 		//the IJ1 image operates always on the full image
-		//if (selectedModel.getName().equals(EfficientSAM.FULL_NAME)) {
 		boolean isColorRGB = activeImage.getType() == ImagePlus.COLOR_RGB;
 		Img<?> image = ImageJFunctions.wrap(isColorRGB ? CompositeConverter.makeComposite(activeImage) : activeImage);
-		return Cast.unchecked(Views.permute(image, 0, 1));
-		//} else {
-		//	return Cast.unchecked(ImageJFunctions.wrap(activeImage));
-		//}
+		//return Cast.unchecked(Views.permute(image, 0, 1));
+		return Cast.unchecked(image);
 	}
 
 	@Override