Skip to content

Commit

Permalink
changed back to HYPVINN_SEG_NAME
Browse files Browse the repository at this point in the history
  • Loading branch information
taha-abdullah committed Jun 4, 2024
1 parent 79dc385 commit 9a6822b
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions HypVINN/run_prediction.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,6 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# IMPORTS
import os.path
from typing import TYPE_CHECKING, Optional, cast, Literal
import argparse
from pathlib import Path
Expand All @@ -32,7 +31,6 @@
load_checkpoint_config_defaults,
)
from FastSurferCNN.utils.common import assert_no_root, SerialExecutor
from FastSurferCNN.utils.parser_defaults import FASTSURFER_ROOT

from HypVINN.config.hypvinn_files import HYPVINN_SEG_NAME
from HypVINN.data_loader.data_utils import hypo_map_label2subseg, rescale_image
Expand All @@ -47,8 +45,6 @@
from HypVINN.utils.stats_utils import compute_stats
from HypVINN.utils.visualization_utils import plot_qc_images

_doc_HYPVINN_SEG_NAME = os.path.relpath(HYPVINN_SEG_NAME, FASTSURFER_ROOT)

logger = logging.get_logger(__name__)

##
Expand Down Expand Up @@ -211,8 +207,8 @@ def main(
The path to the coronal configuration file.
cfg_sag : Path
The path to the sagittal configuration file.
hypo_segfile : str, default="{_doc_HYPVINN_SEG_NAME}"
The name of the hypothalamus segmentation file. Default is {_doc_HYPVINN_SEG_NAME}.
hypo_segfile : str, default="{HYPVINN_SEG_NAME}"
The name of the hypothalamus segmentation file. Default is {HYPVINN_SEG_NAME}.
allow_root : bool, default=False
Whether to allow running as root user. Default is False.
qc_snapshots : bool, optional
Expand Down

0 comments on commit 9a6822b

Please sign in to comment.