距离正则化水平集MATLAB代码

以下是距离正则化水平集的MATLAB代码：

function [phi, iteration] = distanceRegularizedLevelSet(img, phi_0, mu, lambda, alpha, epsilon, timestep, iter_inner, iter_outer, potentialFunction)
% DISTANCEREGULARIZEDLEVELSET computes the distance regularized level set evolution
% implemented via the split Bregman iterative solver
%
% [phi, iteration] = distanceRegularizedLevelSet(img, phi_0, mu, lambda, alpha, epsilon, timestep, iter_inner, iter_outer, potentialFunction)
%
% Input:
% img: input image
% phi_0: initial level set function
% mu: weight of distance regularization term
% lambda: weight of the weighted length term
% alpha: parameter that specifies the width of the Dirac Delta function
% epsilon: parameter that specifies the width of the Heaviside function
% timestep: time step
% iter_inner: number of inner iterations
% iter_outer: number of outer iterations
% potentialFunction: choice of potential function in the Chan-Vese model
%                    including 'single-well' and 'double-well'
%
% Output:
% phi: final level set function
% iteration: number of iterations
%
% Author: Chunming Li, all rights reserved
% E-mail: li_chunming@hotmail.com   
% URL: http://www.engr.uconn.edu/~cmli/

img = double(img(:,:,1)); % convert to double
phi = phi_0;
contour = zeros(size(phi)); % contour initialization
iteration = 0;
for n=1:iter_outer
    phi = NeumannBoundCond(phi);
    [phi, iteration] = lse_inner_iteration(phi, img, mu, lambda, alpha, epsilon, timestep, iter_inner, potentialFunction, iteration);
    contour = computeContour(phi, 0);
end

该函数调用了另外两个函数：NeumannBoundCond和lse_inner_iteration。NeumannBoundCond函数用于处理边界条件，lse_inner_iteration函数用于执行内部迭代。

完整的MATLAB代码可以在以下网址中找到：http://www.engr.uconn.edu/~cmli/code/level_set/distance_regularized_level_set/level_set.zip