vista2d net from monai component

Signed-off-by: am <am>
Project-MONAI · Aug 23, 2024 · 43a84e7 · 43a84e7
1 parent dfe409c
commit 43a84e7
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 82 deletions.
diff --git a/vista2d/configs/hyper_parameters.yaml b/vista2d/configs/hyper_parameters.yaml
@@ -83,7 +83,7 @@ infer:
 
 device: "$torch.device(('cuda:' + os.environ.get('LOCAL_RANK', '0')) if torch.cuda.is_available() else 'cpu')"
 network_def:
-  _target_: scripts.cell_sam_wrapper.CellSamWrapper
+  _target_: monai.networks.nets.cell_sam_wrapper.CellSamWrapper
   checkpoint: $os.path.join(@ckpt_path, "sam_vit_b_01ec64.pth")
 network: $@network_def.to(@device)
 

diff --git a/vista2d/configs/inference.json b/vista2d/configs/inference.json
@@ -20,7 +20,7 @@
     "use_amp": true,
     "amp_dtype": "$torch.float",
     "network_def": {
-        "_target_": "scripts.cell_sam_wrapper.CellSamWrapper",
+        "_target_": "monai.networks.nets.cell_sam_wrapper.CellSamWrapper",
         "checkpoint": "@sam_ckpt_path"
     },
     "network": "$@network_def.to(@device)",

diff --git a/vista2d/scripts/cell_sam_wrapper.py b/vista2d/scripts/cell_sam_wrapper.py