improve logic a little bit

segment-anything-models-java · Oct 29, 2024 · 46f65b0 · 46f65b0
1 parent 88b700d
commit 46f65b0
Showing 1 changed file with 2 additions and 2 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
@@ -489,9 +489,9 @@ else if (collectedPoints.size() > 1 && collectedPoints.size() < 50) {
 						collectedPoints = newCollectedPoints;
 					} else if (collectedPoints.size() > 50) {
 						List<Localizable> newCollectedPoints = new ArrayList<Localizable>();
-						while (newCollectedPoints.size() > 0) {
+						while (newCollectedPoints.size() < 10) {
 							for (Localizable pp : collectedPoints) {
-								if (Math.random() < 0.1) newCollectedPoints.add(pp);
+								if (Math.random() < Math.min(0.1, 50.0 / collectedPoints.size())) newCollectedPoints.add(pp);
 							}
 						}
 						collectedPoints = newCollectedPoints;