Skip to content

Commit 6546459

Browse files
committed
Refactor optimization cmd app
1 parent a2bcfad commit 6546459

File tree

3 files changed

+141
-100
lines changed

3 files changed

+141
-100
lines changed

Modules/DiffusionImaging/FiberTracking/Algorithms/itkFitFibersToImageFilter.cpp

Lines changed: 17 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -466,7 +466,21 @@ void FitFibersToImageFilter::GenerateData()
466466
minimizer.set_lower_bound(l);
467467
minimizer.set_projected_gradient_tolerance(m_GradientTolerance);
468468

469-
MITK_INFO << "Regularization type: " << m_Regularization;
469+
if (m_Regularization==VnlCostFunction::REGU::MSM)
470+
MITK_INFO << "Regularization type: MSM";
471+
else if (m_Regularization==VnlCostFunction::REGU::VARIANCE)
472+
MITK_INFO << "Regularization type: VARIANCE";
473+
else if (m_Regularization==VnlCostFunction::REGU::LASSO)
474+
MITK_INFO << "Regularization type: LASSO";
475+
else if (m_Regularization==VnlCostFunction::REGU::VOXEL_VARIANCE)
476+
MITK_INFO << "Regularization type: VOXEL_VARIANCE";
477+
else if (m_Regularization==VnlCostFunction::REGU::GROUP_LASSO)
478+
MITK_INFO << "Regularization type: GROUP_LASSO";
479+
else if (m_Regularization==VnlCostFunction::REGU::GROUP_VARIANCE)
480+
MITK_INFO << "Regularization type: GROUP_VARIANCE";
481+
else if (m_Regularization==VnlCostFunction::REGU::NONE)
482+
MITK_INFO << "Regularization type: NONE";
483+
470484
if (m_Regularization!=VnlCostFunction::REGU::NONE) // REMOVE FOR NEW FIT AND SET cost.m_Lambda = m_Lambda
471485
{
472486
MITK_INFO << "Estimating regularization";
@@ -493,6 +507,8 @@ void FitFibersToImageFilter::GenerateData()
493507
}
494508
}
495509
}
510+
else
511+
cost.m_Lambda = 0;
496512
MITK_INFO << "Using regularization factor of " << cost.m_Lambda << " (λ: " << m_Lambda << ")";
497513

498514
MITK_INFO << "Fitting fibers";

Modules/DiffusionImaging/FiberTracking/Algorithms/itkTdiToVolumeFractionFilter.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ void TdiToVolumeFractionFilter< TPixelType >::ThreadedGenerateData(const OutputI
120120
else if (comp_sum>1)
121121
intra_ax_val = 1.0 - (csf_val+gm_val);
122122
if (intra_ax_val<-0.01)
123-
mitkThrow() << "Corrupted volume fraction. intra_ax_val=" << intra_ax_val;
123+
MITK_INFO << "Corrupted volume fraction. intra_ax_val=" << intra_ax_val;
124124
if (intra_ax_val<0)
125125
intra_ax_val = 0;
126126
}

0 commit comments

Comments
 (0)