Source code for ewoksndreg.intensities.types
from enum import Enum
[docs]
class SitkMetricType(str, Enum):
MeanSquares = "MeanSquares"
CrossCorrelation = "CrossCorrelation"
ANTSNeighborhoodCorrelation = "ANTSNeighborhoodCorrelation"
JointHistogramMutualInformation = "JointHistogramMutualInformation"
MattesMutualInformation = "MattesMutualInformation"
[docs]
class SitkOptimizerType(str, Enum):
Exhaustive = "Exhaustive"
Powell = "Powell"
Evolutionary = "Evolutionary"
GradientDescent = "GradientDescent"
GradientDescentLineSearch = "GradientDescentLineSearch"
CGLineSearch = "CGLineSearch"
LBFGSB = "LBFGSB"
[docs]
class KorniaMetricType(str, Enum):
MeanAbsoluteError = "MeanAbsoluteError"
MeanSquaredError = "MeanSquaredError"
MutualInformation = "MutualInformation"
[docs]
class KorniaOptimizerType(str, Enum):
Adam = "Adam"
SGD = "SGD"
RMSprop = "RMSprop"