Commit 8af4a479 authored by Constantin Pape's avatar Constantin Pape

Add pre-trained DSB model

parent af9495db
......@@ -56,3 +56,8 @@ submit_slurm train_stardist_2d /path/to/data /path/to/model
```
Scripts to train and predict with a 3d stardist model are also available: `train_stardist_3d`, `predict_stardist_3d`. They and can be used in the same way as the scripts for 2d.
## Pretrained models
The folder `models` contains a model pretrained on DAPI, [originally from the stardist repository](https://github.com/mpicbg-csbd/stardist/tree/master/models/paper/2D_dsb2018).
{"n_channel_in": 1, "n_rays": 32, "unet_n_depth": 3, "unet_kernel_size": [3, 3], "unet_n_filter_base": 32, "net_input_shape": [null, null, 1], "train_dist_loss": "mae", "train_epochs": 100, "train_steps_per_epoch": 400, "train_learning_rate": 0.0003, "train_batch_size": 4, "train_tensorboard": true, "train_checkpoint": "weights_best.h5", "train_reduce_lr": {"factor": 0.5, "patience": 10}}
\ No newline at end of file
{"prob": 0.4178193636640508, "nms": 0.5}
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment