otx.algorithms.anomaly.configs.base#

Base configurable parameter for anomaly tasks.

Classes

`PadimAnomalyBaseConfig`([visible_in_ui, ...])	Configurable parameters for PADIM anomaly classification task.
`STFPMAnomalyBaseConfig`([visible_in_ui, ...])	Configurable parameters for STFPM anomaly base task.
`DraemAnomalyBaseConfig`([visible_in_ui, ...])	Configurable parameters for DRAEM anomaly classification task.

class otx.algorithms.anomaly.configs.base.DraemAnomalyBaseConfig(visible_in_ui: bool = True, dataset: DatasetParameters = _Nothing.NOTHING, pot_parameters: POTParameters = _Nothing.NOTHING, nncf_optimization: NNCFOptimization = _Nothing.NOTHING, learning_parameters: LearningParameters = _Nothing.NOTHING, *, id: str | ID | None = ID(), header: str = 'Configuration for Draem', description: str = 'Configuration for Draem')[source]#

Bases: BaseAnomalyConfig

Configurable parameters for DRAEM anomaly classification task.

Method generated by attrs for class DraemAnomalyBaseConfig.

class LearningParameters(visible_in_ui: bool = True, train_batch_size: int = 8, lr: float = 0.0001, early_stopping: EarlyStoppingParameters = _Nothing.NOTHING, max_epochs: int = 700, *, header: str = 'Learning Parameters', description: str = 'Learning Parameters')[source]#

Bases: LearningParameters

Parameters that can be tuned using HPO.

Method generated by attrs for class DraemAnomalyBaseConfig.LearningParameters.

class EarlyStoppingParameters(visible_in_ui: bool = True, metric: str | ConfigurableEnumTypeVar = EarlyStoppingMetrics.IMAGE_ROC_AUC, patience: int = 20, *, header: str = 'Early Stopping Parameters', description: str = 'Early Stopping Parameters')[source]#

Bases: ParameterGroup

Early stopping parameters.

Method generated by attrs for class DraemAnomalyBaseConfig.LearningParameters.EarlyStoppingParameters.

class otx.algorithms.anomaly.configs.base.PadimAnomalyBaseConfig(visible_in_ui: bool = True, dataset: DatasetParameters = _Nothing.NOTHING, pot_parameters: POTParameters = _Nothing.NOTHING, nncf_optimization: NNCFOptimization = _Nothing.NOTHING, learning_parameters: LearningParameters = _Nothing.NOTHING, *, id: str | ID | None = ID(), header: str = 'Configuration for Padim', description: str = 'Configuration for Padim')[source]#

Bases: BaseAnomalyConfig

Configurable parameters for PADIM anomaly classification task.

Method generated by attrs for class PadimAnomalyBaseConfig.

class LearningParameters(visible_in_ui: bool = True, train_batch_size: int = 32, backbone: str | ConfigurableEnumTypeVar = ModelBackbone.RESNET18, *, header: str = 'Learning Parameters', description: str = 'Learning Parameters')[source]#

Bases: LearningParameters

Parameters that can be tuned using HPO.

Method generated by attrs for class PadimAnomalyBaseConfig.LearningParameters.

class otx.algorithms.anomaly.configs.base.STFPMAnomalyBaseConfig(visible_in_ui: bool = True, dataset: DatasetParameters = _Nothing.NOTHING, pot_parameters: POTParameters = _Nothing.NOTHING, nncf_optimization: NNCFOptimization = _Nothing.NOTHING, learning_parameters: LearningParameters = _Nothing.NOTHING, *, id: str | ID | None = ID(), header: str = 'Configuration for STFPM', description: str = 'Configuration for STFPM')[source]#

Bases: BaseAnomalyConfig

Configurable parameters for STFPM anomaly base task.

Method generated by attrs for class STFPMAnomalyBaseConfig.

class LearningParameters(visible_in_ui: bool = True, train_batch_size: int = 32, lr: float = 0.4, momentum: float = 0.9, weight_decay: float = 0.0001, backbone: str | ConfigurableEnumTypeVar = ModelBackbone.RESNET18, early_stopping: EarlyStoppingParameters = _Nothing.NOTHING, max_epochs: int = 100, *, header: str = 'Learning Parameters', description: str = 'Learning Parameters')[source]#

Bases: LearningParameters

Parameters that can be tuned using HPO.

Method generated by attrs for class STFPMAnomalyBaseConfig.LearningParameters.

class EarlyStoppingParameters(visible_in_ui: bool = True, metric: str | ConfigurableEnumTypeVar = EarlyStoppingMetrics.IMAGE_F1, patience: int = 10, *, header: str = 'Early Stopping Parameters', description: str = 'Early Stopping Parameters')[source]#

Bases: ParameterGroup

Early stopping parameters.

Method generated by attrs for class STFPMAnomalyBaseConfig.LearningParameters.EarlyStoppingParameters.