├── config ├── pretraining_cremic.yaml ├── pretraining_snemi3d.yaml ├── seg_snemi3d_d5_1024_u200.yaml └── seg_snemi3d_d5_1024_u200_ema.yaml ├── data ├── cremi │ └── readme.md ├── kasthuri15 │ └── readme.md └── snemi3d │ └── readme.md ├── dataloader ├── data_provider_labeled.py ├── data_provider_pretraining.py ├── data_provider_unlabel.py ├── data_provider_unlabel_ema.py ├── provider_valid.py └── provider_valid_pretraining.py ├── evaluate.py ├── evaluate_lmc.py ├── evaluate_lmc_models.py ├── evaluate_mala_models.py ├── evaluate_waterz.py ├── images ├── id1455.png ├── id5884.png └── id6913.png ├── inference.py ├── loss ├── loss.py ├── loss_malis.py └── loss_unlabel.py ├── main.py ├── main_ema.py ├── model ├── basic.py ├── model_para.py ├── model_superhuman.py ├── monitor.py ├── residual.py ├── squeeze_excite.py └── unet3d_mala.py ├── pre_training.py ├── readme.md ├── requirements.txt ├── trained_model ├── pretraining_snemi3d │ └── readme.md ├── seg_ac3_d5_1024_u200_EMA │ └── readme.md ├── seg_ac3_d5_1024_u200_WS │ └── readme.md ├── seg_kasthuri15_d10 │ └── readme.md ├── seg_kasthuri15_d100 │ └── readme.md ├── seg_kasthuri15_d100_u1000 │ └── readme.md ├── seg_kasthuri15_d100_u200 │ └── readme.md ├── seg_kasthuri15_d10_u200 │ └── readme.md └── seg_kasthuri15_d50_u200 │ └── readme.md └── utils ├── aff_util.py ├── affine.py ├── augmentation.py ├── augmentation_affine.py ├── compute_sdf.py ├── consistency_aug.py ├── consistency_aug_perturbations.py ├── coordinate.py ├── encoder_dict.py ├── flow_display.py ├── flow_synthesis.py ├── gen_pseudo.py ├── image_warp.py ├── malis_loss.py ├── optim_weight_ema.py ├── post_func.py ├── post_lmc.py ├── post_waterz.py ├── seeds_func.py ├── seg_util.py ├── show.py ├── torch_utils.py └── utils.py /config/pretraining_cremic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/config/pretraining_cremic.yaml -------------------------------------------------------------------------------- /config/pretraining_snemi3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/config/pretraining_snemi3d.yaml -------------------------------------------------------------------------------- /config/seg_snemi3d_d5_1024_u200.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/config/seg_snemi3d_d5_1024_u200.yaml -------------------------------------------------------------------------------- /config/seg_snemi3d_d5_1024_u200_ema.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/config/seg_snemi3d_d5_1024_u200_ema.yaml -------------------------------------------------------------------------------- /data/cremi/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/data/cremi/readme.md -------------------------------------------------------------------------------- /data/kasthuri15/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/data/kasthuri15/readme.md -------------------------------------------------------------------------------- /data/snemi3d/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/data/snemi3d/readme.md -------------------------------------------------------------------------------- /dataloader/data_provider_labeled.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/dataloader/data_provider_labeled.py -------------------------------------------------------------------------------- /dataloader/data_provider_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/dataloader/data_provider_pretraining.py -------------------------------------------------------------------------------- /dataloader/data_provider_unlabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/dataloader/data_provider_unlabel.py -------------------------------------------------------------------------------- /dataloader/data_provider_unlabel_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/dataloader/data_provider_unlabel_ema.py -------------------------------------------------------------------------------- /dataloader/provider_valid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/dataloader/provider_valid.py -------------------------------------------------------------------------------- /dataloader/provider_valid_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/dataloader/provider_valid_pretraining.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/evaluate.py -------------------------------------------------------------------------------- /evaluate_lmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/evaluate_lmc.py -------------------------------------------------------------------------------- /evaluate_lmc_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/evaluate_lmc_models.py -------------------------------------------------------------------------------- /evaluate_mala_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/evaluate_mala_models.py -------------------------------------------------------------------------------- /evaluate_waterz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/evaluate_waterz.py -------------------------------------------------------------------------------- /images/id1455.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/images/id1455.png -------------------------------------------------------------------------------- /images/id5884.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/images/id5884.png -------------------------------------------------------------------------------- /images/id6913.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/images/id6913.png -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/inference.py -------------------------------------------------------------------------------- /loss/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/loss/loss.py -------------------------------------------------------------------------------- /loss/loss_malis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/loss/loss_malis.py -------------------------------------------------------------------------------- /loss/loss_unlabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/loss/loss_unlabel.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/main.py -------------------------------------------------------------------------------- /main_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/main_ema.py -------------------------------------------------------------------------------- /model/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/model/basic.py -------------------------------------------------------------------------------- /model/model_para.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/model/model_para.py -------------------------------------------------------------------------------- /model/model_superhuman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/model/model_superhuman.py -------------------------------------------------------------------------------- /model/monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/model/monitor.py -------------------------------------------------------------------------------- /model/residual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/model/residual.py -------------------------------------------------------------------------------- /model/squeeze_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/model/squeeze_excite.py -------------------------------------------------------------------------------- /model/unet3d_mala.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/model/unet3d_mala.py -------------------------------------------------------------------------------- /pre_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/pre_training.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/requirements.txt -------------------------------------------------------------------------------- /trained_model/pretraining_snemi3d/readme.md: -------------------------------------------------------------------------------- 1 | pretraining_snemi3d.ckpt is put here. 2 | 3 | -------------------------------------------------------------------------------- /trained_model/seg_ac3_d5_1024_u200_EMA/readme.md: -------------------------------------------------------------------------------- 1 | seg_ac3_d5_1024_u200_EMA.ckpt is put here. 2 | 3 | -------------------------------------------------------------------------------- /trained_model/seg_ac3_d5_1024_u200_WS/readme.md: -------------------------------------------------------------------------------- 1 | seg_ac3_d5_1024_u200_WS.ckpt is put here. 2 | 3 | -------------------------------------------------------------------------------- /trained_model/seg_kasthuri15_d10/readme.md: -------------------------------------------------------------------------------- 1 | seg_kasthuri15_d10.ckpt is put here. 2 | 3 | -------------------------------------------------------------------------------- /trained_model/seg_kasthuri15_d100/readme.md: -------------------------------------------------------------------------------- 1 | seg_kasthuri15_d100.ckpt is put here. 2 | 3 | -------------------------------------------------------------------------------- /trained_model/seg_kasthuri15_d100_u1000/readme.md: -------------------------------------------------------------------------------- 1 | seg_kasthuri15_d100_u1000.ckpt is put here. 2 | 3 | -------------------------------------------------------------------------------- /trained_model/seg_kasthuri15_d100_u200/readme.md: -------------------------------------------------------------------------------- 1 | seg_kasthuri15_d100_u200.ckpt is put here. 2 | 3 | -------------------------------------------------------------------------------- /trained_model/seg_kasthuri15_d10_u200/readme.md: -------------------------------------------------------------------------------- 1 | seg_kasthuri15_d10_u200.ckpt is put here. 2 | 3 | -------------------------------------------------------------------------------- /trained_model/seg_kasthuri15_d50_u200/readme.md: -------------------------------------------------------------------------------- 1 | seg_kasthuri15_d50_u200.ckpt is put here. 2 | 3 | -------------------------------------------------------------------------------- /utils/aff_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/aff_util.py -------------------------------------------------------------------------------- /utils/affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/affine.py -------------------------------------------------------------------------------- /utils/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/augmentation.py -------------------------------------------------------------------------------- /utils/augmentation_affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/augmentation_affine.py -------------------------------------------------------------------------------- /utils/compute_sdf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/compute_sdf.py -------------------------------------------------------------------------------- /utils/consistency_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/consistency_aug.py -------------------------------------------------------------------------------- /utils/consistency_aug_perturbations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/consistency_aug_perturbations.py -------------------------------------------------------------------------------- /utils/coordinate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/coordinate.py -------------------------------------------------------------------------------- /utils/encoder_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/encoder_dict.py -------------------------------------------------------------------------------- /utils/flow_display.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/flow_display.py -------------------------------------------------------------------------------- /utils/flow_synthesis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/flow_synthesis.py -------------------------------------------------------------------------------- /utils/gen_pseudo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/gen_pseudo.py -------------------------------------------------------------------------------- /utils/image_warp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/image_warp.py -------------------------------------------------------------------------------- /utils/malis_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/malis_loss.py -------------------------------------------------------------------------------- /utils/optim_weight_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/optim_weight_ema.py -------------------------------------------------------------------------------- /utils/post_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/post_func.py -------------------------------------------------------------------------------- /utils/post_lmc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/post_lmc.py -------------------------------------------------------------------------------- /utils/post_waterz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/post_waterz.py -------------------------------------------------------------------------------- /utils/seeds_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/seeds_func.py -------------------------------------------------------------------------------- /utils/seg_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/seg_util.py -------------------------------------------------------------------------------- /utils/show.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/show.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/weih527/SSNS-Net/HEAD/utils/utils.py --------------------------------------------------------------------------------