go home Home | Main Page | Topics | Namespace List | Class Hierarchy | Alphabetical List | Data Structures | File List | Namespace Members | Data Fields | Globals | Related Pages
Loading...
Searching...
No Matches
ImpactLoss::Dice Class Reference

#include <ImpactLoss.h>

Detailed Description

Binary Dice loss (assumes thresholded activations).

Rounds inputs to {0, 1} before computing overlap.

Definition at line 277 of file ImpactLoss.h.

Inheritance diagram for ImpactLoss::Dice:

Public Member Functions

 Dice ()
void updateValue (torch::Tensor &fixedOutput, torch::Tensor &movingOutput) override
torch::Tensor updateValueAndGetGradientModulator (torch::Tensor &fixedOutput, torch::Tensor &movingOutput) override
Public Member Functions inherited from ImpactLoss::Loss
virtual torch::Tensor GetDerivative (double N) const
virtual double GetValue (double N) const
virtual void initialize (torch::Tensor &output)
 Loss (bool isLossNormalized)
virtual Lossoperator+= (const Loss &other)
void reset ()
void setNumberOfParameters (int numberOfParameters)
void updateDerivativeInJacobianMode (torch::Tensor &jacobian, torch::Tensor &nonZeroJacobianIndices)
virtual void updateValueAndDerivativeInStaticMode (torch::Tensor &fixedOutput, torch::Tensor &movingOutput, torch::Tensor &jacobian, torch::Tensor &nonZeroJacobianIndices)
virtual ~Loss ()=default

Additional Inherited Members

Protected Attributes inherited from ImpactLoss::Loss
torch::Tensor m_Derivative
bool m_Initialized = false
int m_NumberOfParameters
double m_Value

Constructor & Destructor Documentation

◆ Dice()

ImpactLoss::Dice::Dice ( )
inline

Definition at line 280 of file ImpactLoss.h.

Member Function Documentation

◆ updateValue()

void ImpactLoss::Dice::updateValue ( torch::Tensor & fixedOutput,
torch::Tensor & movingOutput )
inlineoverridevirtual

Implements ImpactLoss::Loss.

Definition at line 285 of file ImpactLoss.h.

◆ updateValueAndGetGradientModulator()

torch::Tensor ImpactLoss::Dice::updateValueAndGetGradientModulator ( torch::Tensor & fixedOutput,
torch::Tensor & movingOutput )
inlineoverridevirtual

Implements ImpactLoss::Loss.

Definition at line 315 of file ImpactLoss.h.



Generated on 1774142652 for elastix by doxygen 1.15.0 elastix logo