correct small bug

segment-anything-models-java · Apr 25, 2024 · 7c7eda8 · 7c7eda8
1 parent 2480c69
commit 7c7eda8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/main/java/ai/nets/samj/models/EfficientSamJ.java b/src/main/java/ai/nets/samj/models/EfficientSamJ.java
@@ -393,6 +393,7 @@ protected <T extends RealType<T> & NativeType<T>> void setImageOfInterest(Random
 		long[] dims = rai.dimensionsAsLongArray();
 		if (dims.length == 2)
 			rai = Views.addDimension(rai, 0, 0);
+		dims = rai.dimensionsAsLongArray();
 		if (dims[2] == 1)
 			rai = Views.interval( Views.expandMirrorDouble(rai, new long[] {0, 0, 2}), 
 					Intervals.createMinMax(new long[] {0, 0, 0, dims[0] - 1, dims[1] - 1, 2}) );

diff --git a/src/main/java/ai/nets/samj/models/EfficientViTSamJ.java b/src/main/java/ai/nets/samj/models/EfficientViTSamJ.java
@@ -303,6 +303,7 @@ protected <T extends RealType<T> & NativeType<T>> void setImageOfInterest(Random
 		long[] dims = rai.dimensionsAsLongArray();
 		if (dims.length == 2)
 			rai = Views.addDimension(rai, 0, 0);
+		dims = rai.dimensionsAsLongArray();
 		if (dims[2] == 1)
 			rai = Views.interval( Views.expandMirrorDouble(rai, new long[] {0, 0, 2}), 
 					Intervals.createMinMax(new long[] {0, 0, 0, dims[0] - 1, dims[1] - 1, 2}) );