Mice Uterine Gland Segmentation

Training 2D segmentation model

cd ./root_dir/
python -m mlpipeline.train.run \
  experiment=mcell_sem_seg \
  train.distributed=False \
  model.params.cfg.arch=Unet \
  train.num_epochs=1000 \
  data.batch_size=32 \
  data.num_workers=8

Training 3D segmentation model

cd ./root_dir/
python -m mlpipeline.train.run \
  experiment=mcell3d_sem_seg \
  train.distributed=False \
  model.params.cfg.arch=Unet3D \
  train.num_epochs=1000 \
  data.batch_size=1 \
  data.num_workers=8