Code

Refactor and improvements#121

Code

Merged

wolny merged 14 commits intomasterwolny/pytorch-3dunet:masterfrom imbalanced_labelswolny/pytorch-3dunet:imbalanced_labelsCopy head branch name to clipboardJan 6, 2025

+534-255

Commits

Commits on Oct 4, 2024

• change image logging for multi-class semantic segmentation

  wolny

  committed
  4c0957e

  Copy full SHA for 4c0957e

• make predictor more efficient

  wolny

  committed
  8d3dfb8

  Copy full SHA for 8d3dfb8

Commits on Jan 1, 2025

• skip background class by default when computing IoU

  wolny

  committed
  7d047d4

  Copy full SHA for 7d047d4

Commits on Jan 6, 2025

• avoid double activation during validation, but introducing the forward_logits function

  wolny

  committed
  cf9c7d5

  Copy full SHA for cf9c7d5

• create datasets concurrently; add missing setuptools dependency for conda build

  wolny

  committed
  5cb8e2b

  Copy full SHA for 5cb8e2b

• speed up validation by logging images in a separate thread

  wolny

  committed
  68cbe8f

  Copy full SHA for 68cbe8f

• log percentage of filtered slices

  wolny

  committed
  a9ea0d9

  Copy full SHA for a9ea0d9

• convert weights to float explicitly

  wolny

  committed
  b206908

  Copy full SHA for b206908

• add RandomGammaCorrection augmentation

  wolny

  committed
  438cccc

  Copy full SHA for 438cccc

• add zero-one normalization to ToTensor

  wolny

  committed
  a9c25ec

  Copy full SHA for a9c25ec

• add RandomScaler augmentation

  wolny

  committed
  1c9abf4

  Copy full SHA for 1c9abf4

• update augmentation config for tests

  wolny

  committed
  48e695c

  Copy full SHA for 48e695c

• execute random scale with probability; make gamma correction aug more robust

  wolny

  committed
  c3f18ed

  Copy full SHA for c3f18ed

• update sample configs with shift/scale augmentation

  wolny

  committed
  9551cef

  Copy full SHA for 9551cef