ecandeloro commited on
Commit
3666d6a
·
verified ·
1 Parent(s): fa5342b

Add StarDist segmentation model

Browse files
segmentation/stardist/config.json ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "n_dim": 2,
3
+ "axes": "YXC",
4
+ "n_channel_in": 3,
5
+ "n_channel_out": 33,
6
+ "train_checkpoint": "weights_best.h5",
7
+ "train_checkpoint_last": "weights_last.h5",
8
+ "train_checkpoint_epoch": "weights_now.h5",
9
+ "n_rays": 32,
10
+ "grid": [
11
+ 2,
12
+ 2
13
+ ],
14
+ "backbone": "unet",
15
+ "unet_n_depth": 3,
16
+ "unet_kernel_size": [
17
+ 3,
18
+ 3
19
+ ],
20
+ "unet_n_filter_base": 32,
21
+ "unet_n_conv_per_depth": 2,
22
+ "unet_pool": [
23
+ 2,
24
+ 2
25
+ ],
26
+ "unet_activation": "relu",
27
+ "unet_last_activation": "relu",
28
+ "unet_batch_norm": false,
29
+ "unet_dropout": 0.0,
30
+ "unet_prefix": "",
31
+ "net_conv_after_unet": 128,
32
+ "net_input_shape": [
33
+ null,
34
+ null,
35
+ 3
36
+ ],
37
+ "net_mask_shape": [
38
+ null,
39
+ null,
40
+ 1
41
+ ],
42
+ "train_shape_completion": false,
43
+ "train_completion_crop": 32,
44
+ "train_patch_size": [
45
+ 512,
46
+ 512
47
+ ],
48
+ "train_background_reg": 0.0001,
49
+ "train_foreground_only": 0.9,
50
+ "train_dist_loss": "mae",
51
+ "train_loss_weights": [
52
+ 1,
53
+ 0.1
54
+ ],
55
+ "train_epochs": 200,
56
+ "train_steps_per_epoch": 50,
57
+ "train_learning_rate": 0.0003,
58
+ "train_batch_size": 8,
59
+ "train_n_val_patches": 3,
60
+ "train_tensorboard": true,
61
+ "train_reduce_lr": {
62
+ "factor": 0.5,
63
+ "patience": 20,
64
+ "min_delta": 0
65
+ },
66
+ "use_gpu": true
67
+ }
segmentation/stardist/thresholds.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"prob": 0.6924782541382084, "nms": 0.3}
segmentation/stardist/weights_best.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c7d4af1a53ad6b260f9fff828ef15e7878e9e2b8b4c5667af6f72cc209b4110f
3
+ size 5775632