fixes missing --verbose statement for cellpose 2 in training code

BIOP · Aug 4, 2022 · 9b78185 · 9b78185
1 parent b90099f
commit 9b78185
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/qupath/ext/biop/cellpose/Cellpose2D.java b/src/main/java/qupath/ext/biop/cellpose/Cellpose2D.java
@@ -671,7 +671,8 @@ private void runCellposeTraining() throws IOException, InterruptedException {
 
         if (useGPU) cellposeArguments.add("--use_gpu");
 
-        if (cellposeSetup.getVersion().equals(CellposeSetup.CellposeVersion.CELLPOSE_1))
+        if ( cellposeSetup.getVersion().equals(CellposeSetup.CellposeVersion.CELLPOSE_1) ||
+                cellposeSetup.getVersion().equals(CellposeSetup.CellposeVersion.CELLPOSE_2) )
             cellposeArguments.add("--verbose");
 
         veRunner.setArguments(cellposeArguments);