Skip to content

Commit b34b583

Browse files
committed
add method to know which model type are we wroking with
1 parent 21b3ad1 commit b34b583

File tree

2 files changed

+18
-1
lines changed

2 files changed

+18
-1
lines changed

src/main/java/ai/nets/samj/install/EfficientViTSamEnvManager.java

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ public static EfficientViTSamEnvManager create(String path, String modelType) {
157157
* @param path
158158
* the path where the corresponding micromamba shuold be installed
159159
* @param modelType
160-
* each of the possible model types (sizes) that EfficientiTSAM can have. They are the keys of
160+
* each of the possible model types (sizes) that EfficientViTSAM can have. They are the keys of
161161
* the following map {@link #EFFICIENTVITSAM_BYTE_SIZES_MAP}
162162
* @param consumer
163163
* an specific consumer where info about the installation is going to be communicated
@@ -193,6 +193,15 @@ public static EfficientViTSamEnvManager create(Consumer<String> consumer) {
193193
return create(DEFAULT_DIR, null, consumer);
194194
}
195195

196+
/**
197+
*
198+
* @return which of the possible EfficientViTSAM this is. The possible variants are the
199+
* keys of the following map: {@link #EFFICIENTVITSAM_BYTE_SIZES_MAP}
200+
*/
201+
public String getModelType() {
202+
return this.modelType;
203+
}
204+
196205
/**
197206
* Check whether the Python environment with the corresponding packages needed to run EfficientSAM
198207
* has been installed or not. The environment folder should be named {@value #EVITSAM_ENV_NAME}

src/main/java/ai/nets/samj/install/Sam2EnvManager.java

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -200,6 +200,14 @@ public static Sam2EnvManager create(Consumer<String> consumer) {
200200
return create(DEFAULT_DIR, null, consumer);
201201
}
202202

203+
/**
204+
*
205+
* @return which of the possible SAM2 this is. The possible variants are the keys of the following map: {@link #SAM2_BYTE_SIZES_MAP}
206+
*/
207+
public String getModelType() {
208+
return this.modelType;
209+
}
210+
203211
/**
204212
* Check whether the Python environment with the corresponding packages needed to run EfficientSAM
205213
* has been installed or not. The environment folder should be named {@value #SAM2_ENV_NAME}

0 commit comments

Comments
 (0)