forgot small thing

segment-anything-models-java · Mar 15, 2024 · 734c053 · 734c053
1 parent 1499dc4
commit 734c053
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java b/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java
@@ -148,7 +148,7 @@ public List<Polygon> fetch2dSegmentation(List<Localizable> listOfPoints2D, List<
 					.map(i -> new int[] {(int) i.positionAsDoubleArray()[0], (int) i.positionAsDoubleArray()[1]}).collect(Collectors.toList());
 			List<int[]> negList = listOfNegPoints2D.stream()
 					.map(i -> new int[] {(int) i.positionAsDoubleArray()[0], (int) i.positionAsDoubleArray()[1]}).collect(Collectors.toList());
-			if (negList.size() == 0) return efficientSamJ.processPoints(list);
+			if (negList.size() == 0) return efficientSamJ.processPoints(list, !onlyBiggest);
 			else return efficientSamJ.processPoints(list, negList, !onlyBiggest);
 		} catch (IOException | RuntimeException | InterruptedException e) {
 			log.error(FULL_NAME+", providing empty result because of some trouble: "+e.getMessage());