Changes for saving registered images, minor fixes

NKI-AI · Nov 21, 2024 · 86fd9f4 · 86fd9f4
1 parent f0a2b4f
commit 86fd9f4
Show file tree

Hide file tree

Showing 10 changed files with 4,063 additions and 3,988 deletions.
diff --git a/direct/common/subsample_config.py b/direct/common/subsample_config.py
@@ -1,23 +1,23 @@
-# Copyright (c) DIRECT Contributors
-
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import Optional
-
-from omegaconf import MISSING
-
-from direct.config.defaults import BaseConfig
-from direct.types import MaskFuncMode
-
-
-@dataclass
-class MaskingConfig(BaseConfig):
-    name: str = MISSING
-    accelerations: tuple[float, ...] = (5.0,)
-    center_fractions: Optional[tuple[float, ...]] = (0.1,)
-    uniform_range: bool = False
-    mode: MaskFuncMode = MaskFuncMode.STATIC
-
-    val_accelerations: tuple[float, ...] = (5.0, 10.0)
-    val_center_fractions: Optional[tuple[float, ...]] = (0.1, 0.05)
+# Copyright (c) DIRECT Contributors
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Optional
+
+from omegaconf import MISSING
+
+from direct.config import BaseConfig
+from direct.types import MaskFuncMode
+
+
+@dataclass
+class MaskingConfig(BaseConfig):
+    name: str = MISSING
+    accelerations: tuple[float, ...] = (5.0,)
+    center_fractions: Optional[tuple[float, ...]] = (0.1,)
+    uniform_range: bool = False
+    mode: MaskFuncMode = MaskFuncMode.STATIC
+
+    val_accelerations: tuple[float, ...] = (5.0, 10.0)
+    val_center_fractions: Optional[tuple[float, ...]] = (0.1, 0.05)