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 Inferdimension (
int
) – Input dimensiondescription (
str
) – Descriptionconfig (
Optional
[None
,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 Inferdimension (
int
) – Input dimensiondescription (
str
) – Descriptionconfig (
Optional
[None
,Dict
[str
,Any
]]) – K,V pairs to be part of user config
- 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'¶