UNETR Training with Affinities #166

anwai98 · 2023-11-02T17:13:49Z

Add unetr training script with affinities for livecell dataset

TODO: would be merging to dev instead of main in constantinpape/torch-em

Refactoring the unetr training and inference scripts to adapt to training for all cell-types

anwai98 · 2023-11-08T12:45:24Z

@constantinpape this looks good to go!

anwai98 added 2 commits November 2, 2023 18:10

Add training script for livecell with affinities

648fb14

Update metric in training

1cda926

anwai98 changed the base branch from main to dev November 3, 2023 09:48

anwai98 changed the base branch from dev to main November 3, 2023 09:50

anwai98 changed the base branch from main to dev November 3, 2023 10:06

anwai98 added 4 commits November 3, 2023 16:54

Refactor inference for livecell unetr

625443f

Update mutex watershed segmentation functionality

e4d77b3

Update inference with affinities for unetr

5b4a3ae

Update MWS Evaluation

249b55e

anwai98 marked this pull request as ready for review November 6, 2023 15:16

anwai98 and others added 3 commits November 6, 2023 23:07

Update min_size for watershed

da1edb5

Refactor UNETR Training (#4)

a7a7a4c

Refactoring the unetr training and inference scripts to adapt to training for all cell-types

Update watershed min_size defaults for size filtering

185e4a3

constantinpape approved these changes Nov 8, 2023

View reviewed changes

constantinpape merged commit bd0bc5b into constantinpape:dev Nov 8, 2023
2 checks passed

Provide feedback