monailabel.interfaces.tasks.infer_v2 module

class monailabel.interfaces.tasks.infer_v2.InferTask(type, labels, dimension, description, config=None)[source]

Bases: object

Inference Task

Parameters
  • type (Union[str, InferType]) – Type of Infer (segmentation, deepgrow etc..)

  • labels (Union[str, None, Sequence[str], Dict[Any, Any]]) – Labels associated to this Infer

  • dimension (int) – Input dimension

  • description (str) – Description

  • config (Optional[Dict[str, Any]]) – K,V pairs to be part of user config

__init__(type, labels, dimension, description, config=None)[source]
Parameters
  • type (Union[str, InferType]) – Type of Infer (segmentation, deepgrow etc..)

  • labels (Union[str, None, Sequence[str], Dict[Any, Any]]) – Labels associated to this Infer

  • dimension (int) – Input dimension

  • description (str) – Description

  • config (Optional[Dict[str, Any]]) – K,V pairs to be part of user config

config()[source]
Return type

Dict[str, Any]

get_path(validate=True)[source]
info()[source]
Return type

Dict[str, Any]

abstract is_valid()[source]
Return type

bool

class monailabel.interfaces.tasks.infer_v2.InferType(value)[source]

Bases: str, enum.Enum

Type of Inference Model

SEGMENTATION -            Segmentation Model
ANNOTATION -              Annotation Model
CLASSIFICATION -          Classification Model
DEEPGROW -                Deepgrow Interactive Model
DEEPEDIT -                DeepEdit Interactive Model
SCRIBBLES -               Scribbles Model
DETECTION -               Detection Model
OTHERS -                  Other Model Type
ANNOTATION: str = 'annotation'
CLASSIFICATION: str = 'classification'
DEEPEDIT: str = 'deepedit'
DEEPGROW: str = 'deepgrow'
DETECTION: str = 'detection'
OTHERS: str = 'others'
SCRIBBLES: str = 'scribbles'
SEGMENTATION: str = 'segmentation'