set model to null when closed

segment-anything-models-java · Aug 7, 2024 · e048970 · e048970
1 parent b8ef3e6
commit e048970
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 0 deletions.
diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java b/src/main/java/ai/nets/samj/communication/model/EfficientSAM.java
@@ -227,6 +227,7 @@ public void notifyUiHasBeenClosed() {
 	 */
 	public void closeProcess() {
 		efficientSamJ.close();
+		efficientSamJ = null;
 	}
 
 	@Override

diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML0.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML0.java
@@ -222,6 +222,7 @@ public void notifyUiHasBeenClosed() {
 	 */
 	public void closeProcess() {
 		efficientSamJ.close();
+		efficientSamJ = null;
 	}
 
 	@Override

diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML1.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML1.java
@@ -222,6 +222,7 @@ public void notifyUiHasBeenClosed() {
 	 */
 	public void closeProcess() {
 		efficientSamJ.close();
+		efficientSamJ = null;
 	}
 
 	@Override

diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML2.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML2.java
@@ -221,6 +221,7 @@ public void notifyUiHasBeenClosed() {
 	 */
 	public void closeProcess() {
 		efficientSamJ.close();
+		efficientSamJ = null;
 	}
 
 	@Override

diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL0.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL0.java
@@ -221,6 +221,7 @@ public void notifyUiHasBeenClosed() {
 	 */
 	public void closeProcess() {
 		efficientSamJ.close();
+		efficientSamJ = null;
 	}
 
 	@Override

diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL1.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL1.java
@@ -100,6 +100,8 @@ public boolean isInstalled() {
 	 */
 	public void setImage(final RandomAccessibleInterval<?> image, final SAMJLogger useThisLoggerForIt) 
 			throws IOException, InterruptedException, RuntimeException {
+		if (this.efficientSamJ == null)
+			this.efficientSamJ = EfficientViTSamJ.initializeSam("xl1", manager);
 		try {
 			this.log = useThisLoggerForIt;
 			AbstractSamJ.DebugTextPrinter filteringLogger = text -> {
@@ -223,6 +225,7 @@ public void notifyUiHasBeenClosed() {
 	 */
 	public void closeProcess() {
 		efficientSamJ.close();
+		efficientSamJ = null;
 	}
 
 	@Override