initialize efficientsam if it has been closed

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

Hide file tree

Showing 5 changed files with 11 additions and 2 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
@@ -82,8 +82,7 @@ public class EfficientSAM implements SAMModel {
 	 */
 	public EfficientSAM() throws IOException, RuntimeException, InterruptedException {
 		this.manager = EfficientSamEnvManager.create();
-		efficientSamJ = EfficientSamJ.initializeSam(
-				manager);
+		efficientSamJ = EfficientSamJ.initializeSam(manager);
 	}
 
 	@Override
@@ -116,6 +115,8 @@ public boolean isInstalled() {
 	 */
 	public void setImage(final RandomAccessibleInterval<?> image, final SAMJLogger useThisLoggerForIt) 
 			throws IOException, InterruptedException, RuntimeException {
+		if (this.efficientSamJ == null)
+			efficientSamJ = EfficientSamJ.initializeSam(manager);
 		try {
 			this.log = useThisLoggerForIt;
 			AbstractSamJ.DebugTextPrinter filteringLogger = text -> {

diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML0.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML0.java
@@ -99,6 +99,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("l0", manager);
 		try {
 			this.log = useThisLoggerForIt;
 			AbstractSamJ.DebugTextPrinter filteringLogger = text -> {

diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML1.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML1.java
@@ -99,6 +99,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("l1", manager);
 		try {
 			this.log = useThisLoggerForIt;
 			AbstractSamJ.DebugTextPrinter filteringLogger = text -> {

diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML2.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAML2.java
@@ -98,6 +98,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("l2", manager);
 		try {
 			this.log = useThisLoggerForIt;
 			AbstractSamJ.DebugTextPrinter filteringLogger = text -> {

diff --git a/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL0.java b/src/main/java/ai/nets/samj/communication/model/EfficientViTSAMXL0.java
@@ -98,6 +98,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("xl0", manager);
 		try {
 			this.log = useThisLoggerForIt;
 			AbstractSamJ.DebugTextPrinter filteringLogger = text -> {