File tree 2 files changed +13
-10
lines changed
src/main/java/ch/epfl/biop/wrappers
2 files changed +13
-10
lines changed Original file line number Diff line number Diff line change @@ -154,15 +154,17 @@ public void run() {
154
154
settings .setEnvPath (env_path .toString ());
155
155
settings .setEnvType (env_type );
156
156
settings .setDatasetDir (cellposeTempDir .toString ());
157
-
158
- if ( model ==null || model .trim ().isEmpty ()){
159
- System . out . println ("Using custom model" );
160
- model = model_path .toString ();
157
+ String model_used = model ;
158
+ if (model ==null || model .trim ().isEmpty ()){
159
+ ls . info ("Using cellpose custom model: " + model_path );
160
+ model_used = model_path .toString ();
161
161
}
162
- settings .setModel (model );
162
+ settings .setModel (model_used );
163
163
settings .setDiameter (diameter );
164
164
settings .setChannel1 (ch1 );
165
- if (ch2 > -1 ) settings .setChannel2 (ch2 );
165
+ if (ch2 > -1 ) {
166
+ settings .setChannel2 (ch2 );
167
+ }
166
168
settings .setAdditionalFlags (additional_flags );
167
169
168
170
// settings are done , so we can now process the imp with cellpose
Original file line number Diff line number Diff line change @@ -157,11 +157,12 @@ public void run() {
157
157
settings .setEnvType (env_type );
158
158
settings .setDatasetDir (omniposeTempDir .toString ());
159
159
160
- if ( model ==null || model .trim ().isEmpty ()){
161
- System .out .println ("Using custom model" );
162
- model = model_path .toString ();
160
+ String model_used = model ;
161
+ if (model ==null || model .trim ().isEmpty ()){
162
+ ls .info ("Using omnipose custom model: " +model_path );
163
+ model_used = model_path .toString ();
163
164
}
164
- settings .setModel (model );
165
+ settings .setModel (model_used );
165
166
settings .setDiameter (diameter );
166
167
settings .setChannel1 (ch1 );
167
168
if (ch2 > -1 ) settings .setChannel2 (ch2 );
You can’t perform that action at this time.
0 commit comments