search

teamtomo / membrain-seg

membrane segmentation in 3D for cryo-ET
Other
47 stars 12 forks source link

Misc features #36

Closed LorenzLamm closed 8 months ago

LorenzLamm commented 9 months ago

Hi,

I have implemented a couple of potentially useful features from some user requests, and packed it together with some small bug fixes and documentation:

  1. Adjustable thresholding: It's now possible to set a different threshold for the scoremap during inference (instead of default 0.0)
  2. Related to this: I added the functionality "membrain thresholds" that takes as input a scoremap and applied the given thresholds to it
  3. optional deactivation of test time augmentation: It is now possible to deactivate test time augmentation. Test time augmentation helps a bit with the segmentation performance, but also increases computation 8-fold. I guess it can be useful for larger datasets to deactivate it
  4. documentation for the above functionalities
  5. fixed some typos in the documentation

I think most things should be relatively straightforward, but feel free take a look and let me know if you spot anything :)