File size: 708 Bytes
be2715b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from segmentation_2d.zero_shot_SAM_2d import zero_shot_sam_2d
from segmentation_2d.zero_shot_LVMMed_SAM_2d import zero_shot_lvmmed_sam_2d
from utils.func import (
parse_config,
load_config
)
if __name__=="__main__":
yml_args = parse_config()
cfg = load_config(yml_args.config)
assert cfg.base.is_2D + cfg.base.is_3D == 1
if cfg.base.is_3D:
assert NotImplementedError(f"[Error] We have not yet implemented this function for 3D dataset. You could try implement this similarly based on our 3D implementation for MedSAM.")
if yml_args.lvm_med_encoder_path != '':
zero_shot_lvmmed_sam_2d(yml_args, cfg)
else:
zero_shot_sam_2d(yml_args, cfg) |