monailabel.sam2.infer module

class monailabel.sam2.infer.ImageCache[source]

Bases: object

cleanup()[source]
monitor()[source]
class monailabel.sam2.infer.Sam2InferTask(model_dir, type=InferType.ANNOTATION, dimension=2, labels=None, additional_info=None, image_loader=<monai.transforms.io.dictionary.LoadImaged object>, post_trans=None, writer=<monailabel.transform.writer.Writer object>, config=None)[source]

Bases: monailabel.interfaces.tasks.infer_v2.InferTask

Parameters

  • type – Type of Infer (segmentation, deepgrow etc..)

  • labels – Labels associated to this Infer

  • dimension – Input dimension

  • description – Description

  • config – K,V pairs to be part of user config

info()[source]
Return type

Dict[str, Any]

is_valid()[source]
Return type

bool

run2d(image_tensor, request, debug=False)[source]
run_3d(image_tensor, set_image_state, request, debug=False)[source]

previous

monailabel.sam2 package

next

monailabel.sam2.utils module