Skip to content

Commit

Permalink
add supported apps
Browse files Browse the repository at this point in the history
Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com>
  • Loading branch information
KumoLiu committed Oct 17, 2024
1 parent 0dcdff0 commit e0287db
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 4 deletions.
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
{
"schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
"schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20240725.json",
"version": "0.0.1",
"changelog": {
"0.0.1": "add real time spleen segmentation model"
Expand All @@ -26,6 +26,10 @@
"Xia, Yingda, et al. '3D Semi-Supervised Learning with Uncertainty-Aware Multi-View Co-Training. arXiv preprint arXiv:1811.12506 (2018). https://arxiv.org/abs/1811.12506.",
"Kerfoot E., Clough J., Oksuz I., Lee J., King A.P., Schnabel J.A. (2019) Left-Ventricle Quantification Using Residual U-Net. In: Pop M. et al. (eds) Statistical Atlases and Computational Models of the Heart. Atrial Segmentation and LV Quantification Challenges. STACOM 2018. Lecture Notes in Computer Science, vol 11395. Springer, Cham. https://doi.org/10.1007/978-3-030-12029-0_40"
],
"supported_apps": {
"train": "monai",
"infer": "holoscan"
},
"network_data_format": {
"inputs": {
"image": {
Expand Down
6 changes: 3 additions & 3 deletions models/spleen_ct_segmentation_real_time/scripts/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,8 +37,8 @@ class InferenceWorkflow(PythonicWorkflow):
Test class simulates the bundle workflow defined by Python script directly.
"""

def __init__(self, workflow_type: str = "inference", properties_path: str = "./properties.json"):
super().__init__(workflow_type=workflow_type, properties_path=properties_path)
def __init__(self, workflow_type: str = "inference", config_file: str | None = None, properties_path: str = "./properties.json"):
super().__init__(workflow_type=workflow_type, properties_path=properties_path, config_file=config_file)
# set root log level to INFO and init a evaluation logger, will be used in `StatsHandler`
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
get_logger("eval_log")
Expand Down Expand Up @@ -73,7 +73,7 @@ def initialize(self):
AsDiscreted(keys="pred", argmax=True),
]
)
self.inferer = SlidingWindowInferer(roi_size=(96, 96, 96), sw_batch_size=1, overlap=0)
self.inferer = SlidingWindowInferer(roi_size=self.parser.roi_size, sw_batch_size=1, overlap=0)

def run(self):
data = self.dataset[0]
Expand Down

0 comments on commit e0287db

Please sign in to comment.