diff --git a/src/main/java/ai/nets/samj/install/EfficientViTSamEnvManager.java b/src/main/java/ai/nets/samj/install/EfficientViTSamEnvManager.java index ad7cd64..4a3b646 100644 --- a/src/main/java/ai/nets/samj/install/EfficientViTSamEnvManager.java +++ b/src/main/java/ai/nets/samj/install/EfficientViTSamEnvManager.java @@ -157,7 +157,7 @@ public static EfficientViTSamEnvManager create(String path, String modelType) { * @param path * the path where the corresponding micromamba shuold be installed * @param modelType - * each of the possible model types (sizes) that EfficientiTSAM can have. They are the keys of + * each of the possible model types (sizes) that EfficientViTSAM can have. They are the keys of * the following map {@link #EFFICIENTVITSAM_BYTE_SIZES_MAP} * @param consumer * an specific consumer where info about the installation is going to be communicated @@ -193,6 +193,15 @@ public static EfficientViTSamEnvManager create(Consumer consumer) { return create(DEFAULT_DIR, null, consumer); } + /** + * + * @return which of the possible EfficientViTSAM this is. The possible variants are the + * keys of the following map: {@link #EFFICIENTVITSAM_BYTE_SIZES_MAP} + */ + public String getModelType() { + return this.modelType; + } + /** * Check whether the Python environment with the corresponding packages needed to run EfficientSAM * has been installed or not. The environment folder should be named {@value #EVITSAM_ENV_NAME} diff --git a/src/main/java/ai/nets/samj/install/Sam2EnvManager.java b/src/main/java/ai/nets/samj/install/Sam2EnvManager.java index ac9991e..9d9ecc9 100644 --- a/src/main/java/ai/nets/samj/install/Sam2EnvManager.java +++ b/src/main/java/ai/nets/samj/install/Sam2EnvManager.java @@ -200,6 +200,14 @@ public static Sam2EnvManager create(Consumer consumer) { return create(DEFAULT_DIR, null, consumer); } + /** + * + * @return which of the possible SAM2 this is. The possible variants are the keys of the following map: {@link #SAM2_BYTE_SIZES_MAP} + */ + public String getModelType() { + return this.modelType; + } + /** * Check whether the Python environment with the corresponding packages needed to run EfficientSAM * has been installed or not. The environment folder should be named {@value #SAM2_ENV_NAME}