ImpactLoss::NCC Class Reference

#include <ImpactLoss.h>

Detailed Description

Normalized Cross Correlation loss over feature vectors.

Computes NCC between fixed and moving features across batches. Derivative is accumulated in static mode using full Jacobian tracking.

Definition at line 479 of file ImpactLoss.h.

Inheritance diagram for ImpactLoss::NCC:

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

Private Attributes
torch::Tensor	m_Sdm
torch::Tensor	m_Sf
torch::Tensor	m_Sfdm
torch::Tensor	m_Sff
torch::Tensor	m_Sfm
torch::Tensor	m_Sm
torch::Tensor	m_Smdm
torch::Tensor	m_Smm

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

NCC()

ImpactLoss::NCC::NCC ( )

inline

Definition at line 486 of file ImpactLoss.h.

Member Function Documentation

GetDerivative()

torch::Tensor ImpactLoss::NCC::GetDerivative ( double N ) const

inlineoverridevirtual

Reimplemented from ImpactLoss::Loss.

Definition at line 580 of file ImpactLoss.h.

GetValue()

double ImpactLoss::NCC::GetValue ( double N ) const

inlineoverridevirtual

Reimplemented from ImpactLoss::Loss.

Definition at line 569 of file ImpactLoss.h.

initialize()

void ImpactLoss::NCC::initialize ( torch::Tensor & output )

inlineoverridevirtual

Reimplemented from ImpactLoss::Loss.

Definition at line 491 of file ImpactLoss.h.

operator+=()

NCC & ImpactLoss::NCC::operator+= ( const Loss & other )

inlineoverridevirtual

Reimplemented from ImpactLoss::Loss.

Definition at line 598 of file ImpactLoss.h.

updateValue()

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

inlineoverridevirtual

Implements ImpactLoss::Loss.

Definition at line 505 of file ImpactLoss.h.

updateValueAndDerivativeInStaticMode()

void ImpactLoss::NCC::updateValueAndDerivativeInStaticMode ( torch::Tensor & fixedOutput, torch::Tensor & movingOutput, torch::Tensor & jacobian, torch::Tensor & nonZeroJacobianIndices )

inlineoverridevirtual

Reimplemented from ImpactLoss::Loss.

Definition at line 516 of file ImpactLoss.h.

updateValueAndGetGradientModulator()

torch::Tensor ImpactLoss::NCC::updateValueAndGetGradientModulator ( torch::Tensor & fixedOutput, torch::Tensor & movingOutput )

inlineoverridevirtual

Implements ImpactLoss::Loss.

Definition at line 538 of file ImpactLoss.h.

Member Data Documentation

m_Sdm

torch::Tensor ImpactLoss::NCC::m_Sdm

private

Definition at line 483 of file ImpactLoss.h.

m_Sf

torch::Tensor ImpactLoss::NCC::m_Sf

private

Definition at line 482 of file ImpactLoss.h.

m_Sfdm

torch::Tensor ImpactLoss::NCC::m_Sfdm

private

Definition at line 483 of file ImpactLoss.h.

m_Sff

torch::Tensor ImpactLoss::NCC::m_Sff

private

Definition at line 482 of file ImpactLoss.h.

m_Sfm

torch::Tensor ImpactLoss::NCC::m_Sfm

private

Definition at line 482 of file ImpactLoss.h.

m_Sm

torch::Tensor ImpactLoss::NCC::m_Sm

private

Definition at line 482 of file ImpactLoss.h.

m_Smdm

torch::Tensor ImpactLoss::NCC::m_Smdm

private

Definition at line 483 of file ImpactLoss.h.

m_Smm

torch::Tensor ImpactLoss::NCC::m_Smm

private

Definition at line 482 of file ImpactLoss.h.