Skip to content

Commit 77f2c19

Browse files
committed
input for custom model
1 parent ecc90d7 commit 77f2c19

1 file changed

Lines changed: 8 additions & 3 deletions

File tree

src/main/java/fiji/plugin/appose/cellpose/cp3/CellposeAppose.java

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import java.awt.EventQueue;
88
import java.awt.Font;
99
import java.awt.Window;
10+
import java.io.File;
1011
import java.io.IOException;
1112
import java.net.URL;
1213
import java.nio.charset.StandardCharsets;
@@ -29,6 +30,7 @@
2930
import org.apposed.appose.Service.Task;
3031
import org.apposed.appose.Service.TaskStatus;
3132
import org.scijava.Initializable;
33+
import org.scijava.ItemVisibility;
3234
import org.scijava.command.Command;
3335
import org.scijava.command.DynamicCommand;
3436
import org.scijava.module.DefaultMutableModuleItem;
@@ -61,12 +63,15 @@
6163
public class CellposeAppose extends DynamicCommand implements Initializable
6264
{
6365

64-
@Parameter( choices = {"cyto3", "nuclei", "tissunet", "livecell", "CP", "cyto2", "cyto2_cp3", "tissuenet_cp3",
66+
@Parameter(label= "Cellpose model", choices = {"cyto3", "nuclei", "tissunet", "livecell", "CP", "cyto2", "cyto2_cp3", "tissuenet_cp3",
6567
"livecell_cp3", "yeast_PhC_cp3", "yeast_BF_cp3", "bact_phase_cp3", "bact_fluor_cp3", "deepbacs_cp3",
6668
"neurips_grayscale_cyto2", "TN1", "TN2", "TN3", "LC1", "LC2", "LC3", "LC4", "neurips_cellpose_default",
67-
"neurips_cellpose_transformer"}, description="Choose CP model to run" )
69+
"neurips_cellpose_transformer"}, description="Choose CP model to run")
6870
private String cp_model = "cyto3"; // cellpose model
6971

72+
@Parameter(label = "Custom model", description = "Custom model path, overrides the Cellpose model", style="file", required = false)
73+
private File custom_model = null;
74+
7075
@Parameter( label = "Diameter", min="0", description="Average diameter of a cell/nuclei (in pixels)" )
7176
private int cell_diameter = 30; // cell diameter
7277

@@ -115,7 +120,7 @@ public void initialize() {
115120

116121
is3D = is3d(imp);
117122

118-
123+
119124
List<String> channelChoices = new ArrayList<>();
120125
for (int i = 1; i <= imp.getNChannels(); i++) {
121126
channelChoices.add(String.valueOf(i));

0 commit comments

Comments
 (0)