def masked_transform(blob,
pad_size=None,
noise_fraction=0.5,
l2denoise=True,
gray=False):
mask = su.rndmask(blob.shape, noise_fraction, dtype=blob.dtype)
blobw = blob * mask
if pad_size is not None:
pad = [(pad_size, pad_size), (pad_size, pad_size)] + \
[(0, 0) for _ in range(blob.ndim-2)]
blobw = np.pad(blobw, pad, mode='constant')
mask = np.pad(mask, pad, 'constant')
if l2denoise:
tvl2opt = tvl2.TVL2Denoise.Options({
'Verbose': False,
'MaxMainIter': 200,
'gEvalY': False,
'AutoRho': {
'Enabled': True
},
'DFidWeight': mask
})
denoiser = tvl2.TVL2Denoise(blobw,
0.05,
tvl2opt,
caxis=None if gray else 2)
sl = denoiser.solve()
sh = mask * (blobw - sl)
else:
sl, sh = np.zeros_like(blobw), blobw
return sl, sh, mask
def test_01(self):
lmbda = 3
try:
b = tvl2.TVL2Denoise(self.D, lmbda)
b.solve()
except Exception as e:
print(e)
assert (0)
def test_01(self):
lmbda = 1e-1
opt = tvl2.TVL2Denoise.Options({'Verbose': False, 'gEvalY': False,
'MaxMainIter': 250, 'rho': 10*lmbda})
b = tvl2.TVL2Denoise(self.D, lmbda, opt, axes=(0,1,2))
X = b.solve()
assert(np.abs(b.itstat[-1].ObjFun - 366.04267554965134) < 1e-3)
assert(sm.mse(self.U,X) < 1e-3)
def test_01(self):
lmbda = 1e-1
opt = tvl2.TVL2Denoise.Options({'Verbose': False, 'gEvalY': False,
'MaxMainIter': 300, 'rho': 75*lmbda})
b = tvl2.TVL2Denoise(self.D, lmbda, opt)
X = b.solve()
assert(np.abs(b.itstat[-1].ObjFun - 32.875710674129564) < 1e-3)
assert(sm.mse(self.U,X) < 1e-3)
def test_05(self):
lmbda = 3
dt = np.float64
opt = tvl2.TVL2Denoise.Options({'Verbose': False, 'MaxMainIter': 20,
'AutoRho': {'Enabled': True}, 'DataType': dt})
b = tvl2.TVL2Denoise(self.D, lmbda, opt=opt)
b.solve()
assert(b.X.dtype == dt)
assert(b.Y.dtype == dt)
assert(b.U.dtype == dt)
def test_01(self):
lmbda = 1e-1
opt = tvl2.TVL2Denoise.Options({
'Verbose': False,
'gEvalY': False,
'MaxMainIter': 250,
'rho': 10 * lmbda
})
b = tvl2.TVL2Denoise(self.D, lmbda, opt, axes=(0, 1))
X = b.solve()
assert np.abs(b.itstat[-1].ObjFun - 363.0802047) < 1e-3
assert sm.mse(self.U, X) < 1e-3
def train_models(solvers, train_loader, args):
"""Train for all solvers."""
dname = args.dataset if not args.use_gray else args.dataset+'.gray'
masks = []
shs = []
for e, blob in enumerate(train_loader):
mask = su.rndmask(blob.shape, args.noise_fraction, dtype=blob.dtype)
blobw = blob * mask
if not args.dont_pad_boundary:
pad = [(0, args.patch_size-1), (0, args.patch_size-1)] + \
[(0, 0) for _ in range(blob.ndim-2)]
blobw = np.pad(blobw, pad, 'constant')
mask = np.pad(mask, pad, 'constant')
# l2-TV denoising
tvl2opt = tvl2.TVL2Denoise.Options({
'Verbose': False, 'MaxMainIter': 200, 'gEvalY': False,
'AutoRho': {'Enabled': True}, 'DFidWeight': mask
})
denoiser = tvl2.TVL2Denoise(blobw, args.l2_lambda, tvl2opt,
caxis=None if args.use_gray else 2)
sl = denoiser.solve()
sh = mask * (blobw - sl)
# save masks and sh
masks.append(mask)
shs.append(sh)
# Update solvers
for k, solver in solvers.items():
solver.solve(sh, W=mask)
np.save(os.path.join(args.output_path, k, '{}.{}.npy'.format(dname, e)),
solver.getdict().squeeze())
if args.visdom is not None:
tiled_dict = su.tiledict(solver.getdict().squeeze())
if not args.use_gray:
tiled_dict = tiled_dict.transpose(2, 0, 1)
args.visdom.image(tiled_dict, opts=dict(caption=f'{k}.{e}'))
# snapshot blobs and masks
masks = np.concatenate(masks, axis=-1)
shs = np.concatenate(shs, axis=-1)
np.save(os.path.join(args.output_path, 'train_masks.npy'), masks)
np.save(os.path.join(args.output_path, 'train_blobs.npy'), shs)
return solvers, shs, masks
"""
lmbda = 0.04
opt = tvl2.TVL2Denoise.Options({
'Verbose': True,
'MaxMainIter': 200,
'gEvalY': False,
'AutoRho': {
'Enabled': True
}
})
"""
Create solver object and solve, returning the the denoised image ``imgr``.
"""
b = tvl2.TVL2Denoise(imgn, lmbda, opt)
imgr = b.solve()
"""
Display solve time and denoising performance.
"""
print("TVL2Denoise solve time: %5.2f s" % b.timer.elapsed('solve'))
print("Noisy image PSNR: %5.2f dB" % metric.psnr(img, imgn))
print("Denoised image PSNR: %5.2f dB" % metric.psnr(img, imgr))
"""
Display reference, corrupted, and denoised images.
"""
fig = plot.figure(figsize=(20, 5))
plot.subplot(1, 3, 1)
plot.imview(img, fgrf=fig, title='Reference')
Construct padded mask and test image.
"""
mskp = zpad(msk)
imgwp = spad(imgw)
"""
$\ell_2$-TV denoising with a spatial mask as a non-linear lowpass filter. The highpass component is the difference between the test image and the lowpass component, multiplied by the mask for faster convergence of the convolutional sparse coding (see :cite:`wohlberg-2017-convolutional3`).
"""
lmbda = 0.05
opt = tvl2.TVL2Denoise.Options({'Verbose': False, 'MaxMainIter': 200,
'DFidWeight': mskp, 'gEvalY': False,
'AutoRho': {'Enabled': True}})
b = tvl2.TVL2Denoise(imgwp, lmbda, opt, caxis=2)
sl = b.solve()
sh = mskp * (imgwp - sl)
"""
Load dictionary.
"""
D = util.convdicts()['RGB:8x8x3x64']
"""
Set up :class:`.admm.cbpdn.ConvBPDN` options.
"""
Sw = W * S
"""
$\ell_2$-TV denoising with a spatial mask as a non-linear lowpass filter.
"""
lmbda = 0.1
opt = tvl2.TVL2Denoise.Options({
'Verbose': False,
'MaxMainIter': 200,
'DFidWeight': W,
'gEvalY': False,
'AutoRho': {
'Enabled': True
}
})
b = tvl2.TVL2Denoise(Sw, lmbda, opt)
sl = b.solve()
sh = Sw - sl
"""
CDL without a spatial mask using :class:`.admm.cbpdndl.ConvBPDNDictLearn`.
"""
lmbda = 0.05
opt1 = cbpdndl.ConvBPDNDictLearn.Options({
'Verbose': True,
'MaxMainIter': 200,
'AccurateDFid': True,
'CBPDN': {
'rho': 50.0 * lmbda + 0.5
},
'CCMOD': {
Create random mask and apply to training images.
"""
frc = 0.5
W = util.rndmask(S.shape[0:3] + (1,), frc, dtype=np.float32)
Sw = W * S
"""
$\ell_2$-TV denoising with a spatial mask as a non-linear lowpass filter.
"""
lmbda = 0.1
opt = tvl2.TVL2Denoise.Options({'Verbose': False, 'MaxMainIter': 200,
'DFidWeight': W, 'gEvalY': False, 'AutoRho': {'Enabled': True}})
b = tvl2.TVL2Denoise(Sw, lmbda, opt, caxis=2)
sl = b.solve()
sh = Sw - sl
"""
CDL without a spatial mask using :class:`.dictlrn.cbpdndl.ConvBPDNDictLearn`. (Note that :class:`.prlcnscdl.ConvBPDNMaskDcplDictLearn_Consensus` solves the same problem, but is substantially faster on a multi-core architecture.)
"""
lmbda = 0.05
opt1 = cbpdndl.ConvBPDNDictLearn.Options({'Verbose': True,
'MaxMainIter': 200, 'AccurateDFid': True,
'CBPDN': {'rho': 50.0*lmbda + 0.5},
'CCMOD': {'rho': 3e2}}, dmethod='cns')
d1 = cbpdndl.ConvBPDNDictLearn(D0, sh, lmbda, opt1, dmethod='cns')
D1 = d1.solve()
def gengraphs(pth, nopyfftw):
"""
Generate call graph images when necessary. Parameter pth is the path
to the directory in which images are to be created. Parameter nopyfftw
is a flag indicating whether it is necessary to avoid using pyfftw.
"""
srcmodflt = '^sporco.admm'
srcqnmflt = r'^((?!<locals>|__new|_name_nested).)*$'
dstqnmflt = r'^((?!<locals>|__new|_name_nested).)*$'
fnmsub = ('^sporco.admm.', '')
grpflt = r'^[^\.]*.[^\.]*'
lnksub = (r'^([^\.]*).(.*)',
r'../../sporco.admm.\1.html#sporco.admm.\1.\2')
fntsz = 9
fntfm = 'Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans'
kwargs = {'fntsz': fntsz, 'fntfm': fntfm, 'rmsz': True}
ct = jonga.CallTracer(srcmodflt=srcmodflt,
srcqnmflt=srcqnmflt,
dstqnmflt=dstqnmflt,
fnmsub=fnmsub,
grpflt=grpflt,
lnksub=lnksub)
# Make destination directory if it doesn't exist
if not os.path.exists(pth):
os.makedirs(pth, exist_ok=True)
# Handle environment in which pyfftw is unavailable
if nopyfftw:
import numpy.fft as npfft
import sporco.linalg as spl
def empty(shape, dtype, order='C', n=None):
return np.zeros(shape, dtype=dtype)
spl.pyfftw_empty_aligned = empty
def rfftn_empty(shape, axes, dtype, order='C', n=None):
ashp = list(shape)
raxis = axes[-1]
ashp[raxis] = ashp[raxis] // 2 + 1
cdtype = spl.complex_dtype(dtype)
return np.zeros(ashp, dtype=cdtype)
spl.pyfftw_rfftn_empty_aligned = rfftn_empty
spl.fftn = npfft.fftn
spl.ifftn = npfft.ifftn
spl.rfftn = npfft.rfftn
spl.irfftn = npfft.irfftn
import numpy as np
np.random.seed(12345)
#### bpdn module
from sporco.admm import bpdn
mdnm = 'sporco.admm.bpdn'
D = np.random.randn(8, 16)
s = np.random.randn(8, 1)
lmbda = 0.1
## BPDN class
opt = bpdn.BPDN.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'bpdn_init.svg', **kwargs):
b = bpdn.BPDN(D, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'bpdn_solve.svg', **kwargs):
b.solve()
## BPDNJoint class
opt = bpdn.BPDNJoint.Options({'Verbose': False, 'MaxMainIter': 1})
mu = 0.01
with CallGraph(ct, mdnm, pth, 'bpdnjnt_init.svg', **kwargs):
b = bpdn.BPDNJoint(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'bpdnjnt_solve.svg', **kwargs):
b.solve()
## ElasticNet class
opt = bpdn.ElasticNet.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'elnet_init.svg', **kwargs):
b = bpdn.ElasticNet(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'elnet_solve.svg', **kwargs):
b.solve()
# BPDNProjL1 class
opt = bpdn.BPDNProjL1.Options({'Verbose': False, 'MaxMainIter': 1})
gamma = 2.0
with CallGraph(ct, mdnm, pth, 'bpdnprjl1_init.svg', **kwargs):
b = bpdn.BPDNProjL1(D, s, gamma, opt)
with CallGraph(ct, mdnm, pth, 'bpdnprjl1_solve.svg', **kwargs):
b.solve()
## MinL1InL2Ball class
opt = bpdn.MinL1InL2Ball.Options({'Verbose': False, 'MaxMainIter': 1})
epsilon = 1.0
with CallGraph(ct, mdnm, pth, 'bpdnml1l2_init.svg', **kwargs):
b = bpdn.MinL1InL2Ball(D, s, epsilon, opt)
with CallGraph(ct, mdnm, pth, 'bpdnml1l2_solve.svg', **kwargs):
b.solve()
#### cbpdn module
from sporco.admm import cbpdn
mdnm = 'sporco.admm.cbpdn'
D = np.random.randn(4, 4, 16)
s = np.random.randn(8, 8)
lmbda = 0.1
## ConvBPDN class
opt = cbpdn.ConvBPDN.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'cbpdn_init.svg', **kwargs):
b = cbpdn.ConvBPDN(D, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'cbpdn_solve.svg', **kwargs):
b.solve()
## ConvBPDNJoint class
opt = cbpdn.ConvBPDNJoint.Options({'Verbose': False, 'MaxMainIter': 1})
mu = 0.01
with CallGraph(ct, mdnm, pth, 'cbpdnjnt_init.svg', **kwargs):
b = cbpdn.ConvBPDNJoint(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnjnt_solve.svg', **kwargs):
b.solve()
## ConvElasticNet class
opt = cbpdn.ConvElasticNet.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'celnet_init.svg', **kwargs):
b = cbpdn.ConvElasticNet(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'celnet_solve.svg', **kwargs):
b.solve()
## ConvBPDNGradReg class
opt = cbpdn.ConvBPDNGradReg.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'cbpdngrd_init.svg', **kwargs):
b = cbpdn.ConvBPDNGradReg(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdngrd_solve.svg', **kwargs):
b.solve()
## ConvBPDNProjL1 class
opt = cbpdn.ConvBPDNProjL1.Options({'Verbose': False, 'MaxMainIter': 1})
gamma = 0.5
with CallGraph(ct, mdnm, pth, 'cbpdnprjl1_init.svg', **kwargs):
b = cbpdn.ConvBPDNProjL1(D, s, gamma, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnprjl1_solve.svg', **kwargs):
b.solve()
## ConvMinL1InL2Ball class
opt = cbpdn.ConvMinL1InL2Ball.Options({'Verbose': False, 'MaxMainIter': 1})
epsilon = 0.5
with CallGraph(ct, mdnm, pth, 'cbpdnml1l2_init.svg', **kwargs):
b = cbpdn.ConvMinL1InL2Ball(D, s, epsilon, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnml1l2_solve.svg', **kwargs):
b.solve()
## ConvBPDNMaskDcpl class
opt = cbpdn.ConvBPDNMaskDcpl.Options({'Verbose': False, 'MaxMainIter': 1})
W = np.ones(s.shape)
with CallGraph(ct, mdnm, pth, 'cbpdnmd_init.svg', **kwargs):
b = cbpdn.ConvBPDNMaskDcpl(D, s, lmbda, W, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnmd_solve.svg', **kwargs):
b.solve()
#### cbpdntv module
from sporco.admm import cbpdntv
mdnm = 'sporco.admm.cbpdntv'
D = np.random.randn(4, 4, 16)
s = np.random.randn(8, 8)
lmbda = 0.1
mu = 0.01
## ConvBPDNScalarTV class
opt = cbpdntv.ConvBPDNScalarTV.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'cbpdnstv_init.svg', **kwargs):
b = cbpdntv.ConvBPDNScalarTV(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnstv_solve.svg', **kwargs):
b.solve()
## ConvBPDNVectorTV class
opt = cbpdntv.ConvBPDNVectorTV.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'cbpdnvtv_init.svg', **kwargs):
b = cbpdntv.ConvBPDNVectorTV(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnvtv_solve.svg', **kwargs):
b.solve()
## ConvBPDNRecTV class
opt = cbpdntv.ConvBPDNRecTV.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'cbpdnrtv_init.svg', **kwargs):
b = cbpdntv.ConvBPDNRecTV(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnrtv_solve.svg', **kwargs):
b.solve()
#### cmod module
from sporco.admm import cmod
mdnm = 'sporco.admm.cmod'
X = np.random.randn(8, 16)
S = np.random.randn(8, 16)
## CnstrMOD class
opt = cmod.CnstrMOD.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'cmod_init.svg', **kwargs):
b = cmod.CnstrMOD(X, S, opt=opt)
with CallGraph(ct, mdnm, pth, 'cmod_solve.svg', **kwargs):
b.solve()
#### ccmod module
from sporco.admm import ccmod
mdnm = 'sporco.admm.ccmod'
X = np.random.randn(8, 8, 1, 2, 1)
S = np.random.randn(8, 8, 2)
dsz = (4, 4, 1)
## ConvCnstrMOD_IterSM class
opt = ccmod.ConvCnstrMOD_IterSM.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'ccmodism_init.svg', **kwargs):
b = ccmod.ConvCnstrMOD_IterSM(X, S, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodism_solve.svg', **kwargs):
b.solve()
## ConvCnstrMOD_CG class
opt = ccmod.ConvCnstrMOD_CG.Options({
'Verbose': False,
'MaxMainIter': 1,
'CG': {
'MaxIter': 1
}
})
with CallGraph(ct, mdnm, pth, 'ccmodcg_init.svg', **kwargs):
b = ccmod.ConvCnstrMOD_CG(X, S, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodcg_solve.svg', **kwargs):
b.solve()
## ConvCnstrMOD_Consensus class
opt = ccmod.ConvCnstrMOD_Consensus.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'ccmodcnsns_init.svg', **kwargs):
b = ccmod.ConvCnstrMOD_Consensus(X, S, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodcnsns_solve.svg', **kwargs):
b.solve()
#### ccmodmd module
from sporco.admm import ccmodmd
mdnm = 'sporco.admm.ccmodmd'
X = np.random.randn(8, 8, 1, 2, 1)
S = np.random.randn(8, 8, 2)
W = np.array([1.0])
dsz = (4, 4, 1)
## ConvCnstrMODMaskDcpl_IterSM class
opt = ccmodmd.ConvCnstrMODMaskDcpl_IterSM.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'ccmodmdism_init.svg', **kwargs):
b = ccmodmd.ConvCnstrMODMaskDcpl_IterSM(X, S, W, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodmdism_solve.svg', **kwargs):
b.solve()
## ConvCnstrMODMaskDcpl_CG class
opt = ccmodmd.ConvCnstrMODMaskDcpl_CG.Options({
'Verbose': False,
'MaxMainIter': 1,
'CG': {
'MaxIter': 1
}
})
with CallGraph(ct, mdnm, pth, 'ccmodmdcg_init.svg', **kwargs):
b = ccmodmd.ConvCnstrMODMaskDcpl_CG(X, S, W, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodmdcg_solve.svg', **kwargs):
b.solve()
## ConvCnstrMODMaskDcpl_Consensus class
opt = ccmodmd.ConvCnstrMODMaskDcpl_Consensus.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'ccmodmdcnsns_init.svg', **kwargs):
b = ccmodmd.ConvCnstrMODMaskDcpl_Consensus(X, S, W, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodmdcnsns_solve.svg', **kwargs):
b.solve()
#### bpdndl module
from sporco.admm import bpdndl
mdnm = 'sporco.admm.bpdndl'
D0 = np.random.randn(8, 8)
S = np.random.randn(8, 16)
lmbda = 0.1
## BPDNDictLearn class
opt = bpdndl.BPDNDictLearn.Options({
'Verbose': False,
'MaxMainIter': 1,
'AccurateDFid': True
})
with CallGraph(ct, mdnm, pth, 'bpdndl_init.svg', **kwargs):
b = bpdndl.BPDNDictLearn(D0, S, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'bpdndl_solve.svg', **kwargs):
b.solve()
#### cbpdndl module
from sporco.admm import cbpdndl
mdnm = 'sporco.admm.cbpdndl'
D0 = np.random.randn(4, 4, 16)
s = np.random.randn(8, 8, 10)
lmbda = 0.1
## ConvBPDNDictLearn class
opt = cbpdndl.ConvBPDNDictLearn.Options({
'Verbose': False,
'MaxMainIter': 1,
'AccurateDFid': True
})
with CallGraph(ct, mdnm, pth, 'cbpdndl_init.svg', **kwargs):
b = cbpdndl.ConvBPDNDictLearn(D0, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'cbpdndl_solve.svg', **kwargs):
b.solve()
## ConvBPDNMaskDcplDictLearn class
W = np.array([1.0])
opt = cbpdndl.ConvBPDNMaskDcplDictLearn.Options({
'Verbose': False,
'MaxMainIter': 1,
'AccurateDFid': True
})
with CallGraph(ct, mdnm, pth, 'cbpdnmddl_init.svg', **kwargs):
b = cbpdndl.ConvBPDNMaskDcplDictLearn(D0, s, lmbda, W, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnmddl_solve.svg', **kwargs):
b.solve()
#### tvl1 module
from sporco.admm import tvl1
mdnm = 'sporco.admm.tvl1'
s = np.random.randn(16, 16)
lmbda = 0.1
## TVL1Denoise class
opt = tvl1.TVL1Denoise.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'tvl1den_init.svg', **kwargs):
b = tvl1.TVL1Denoise(s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'tvl1den_solve.svg', **kwargs):
b.solve()
## TVL1Deconv class
opt = tvl1.TVL1Deconv.Options({'Verbose': False, 'MaxMainIter': 1})
h = np.random.randn(3, 3)
with CallGraph(ct, mdnm, pth, 'tvl1dcn_init.svg', **kwargs):
b = tvl1.TVL1Deconv(h, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'tvl1dcn_solve.svg', **kwargs):
b.solve()
#### tvl2 module
from sporco.admm import tvl2
mdnm = 'sporco.admm.tvl2'
s = np.random.randn(16, 16)
lmbda = 0.1
## TVL2Denoise class
opt = tvl2.TVL2Denoise.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'tvl2den_init.svg', **kwargs):
b = tvl2.TVL2Denoise(s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'tvl2den_solve.svg', **kwargs):
b.solve()
## TVL2Deconv class
opt = tvl2.TVL2Deconv.Options({'Verbose': False, 'MaxMainIter': 1})
h = np.random.randn(3, 3)
with CallGraph(ct, mdnm, pth, 'tvl2dcn_init.svg', **kwargs):
b = tvl2.TVL2Deconv(h, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'tvl2dcn_solve.svg', **kwargs):
b.solve()
srcmodflt = '^sporco.fista'
fnmsub = ('^sporco.fista.', '')
lnksub = (r'^([^\.]*).(.*)',
r'../../sporco.fista.\1.html#sporco.fista.\1.\2')
ct = jonga.CallTracer(srcmodflt=srcmodflt,
srcqnmflt=srcqnmflt,
dstqnmflt=dstqnmflt,
fnmsub=fnmsub,
grpflt=grpflt,
lnksub=lnksub)
#### fista.cbpdn module
from sporco.fista import cbpdn
mdnm = 'sporco.fista.cbpdn'
D = np.random.randn(4, 4, 16)
s = np.random.randn(8, 8)
lmbda = 0.1
## ConvBPDN class
opt = cbpdn.ConvBPDN.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'fista_cbpdn_init.svg', **kwargs):
b = cbpdn.ConvBPDN(D, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'fista_cbpdn_solve.svg', **kwargs):
b.solve()
#### fista.ccmod module
from sporco.fista import ccmod
mdnm = 'sporco.fista.ccmod'
X = np.random.randn(8, 8, 1, 2, 1)
S = np.random.randn(8, 8, 2)
dsz = (4, 4, 1)
## ConvCnstrMOD class
opt = ccmod.ConvCnstrMOD.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'ccmodfista_init.svg', **kwargs):
b = ccmod.ConvCnstrMOD(X, S, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodfista_solve.svg', **kwargs):
b.solve()
## ConvCnstrMODMaskDcpl class
opt = ccmod.ConvCnstrMODMask.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'ccmodmdfista_init.svg', **kwargs):
b = ccmod.ConvCnstrMODMask(X, S, W, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodmdfista_solve.svg', **kwargs):
b.solve()
def gengraphs(pth):
"""
Generate call graph images when necessary. Parameter pth is the path
to the directory in which images are to be created.
"""
srcmodflt = '^sporco.admm'
srcqnmflt = r'^((?!<locals>|__new|_name_nested).)*$'
dstqnmflt = r'^((?!<locals>|__new|_name_nested).)*$'
fnmsub = ('^sporco.admm.', '')
grpflt = r'^[^\.]*.[^\.]*'
lnkpfx = '../../modules/'
lnksub = (r'^([^\.]*).([^\.]*)(?:(.__init__|.__call__)|(.[^\.]*))',
lnkpfx + r'sporco.admm.\1.html#sporco.admm.\1.\2\4')
fntsz = 9
fntfm = 'Vera Sans, DejaVu Sans, Liberation Sans, Arial, Helvetica, sans'
kwargs = {'fntsz': fntsz, 'fntfm': fntfm, 'rmsz': True}
ct = jonga.CallTracer(srcmodflt=srcmodflt,
srcqnmflt=srcqnmflt,
dstqnmflt=dstqnmflt,
fnmsub=fnmsub,
grpflt=grpflt,
lnksub=lnksub)
# Make destination directory if it doesn't exist
if not os.path.exists(pth):
os.makedirs(pth, exist_ok=True)
import numpy as np
np.random.seed(12345)
#### bpdn module
from sporco.admm import bpdn
mdnm = 'sporco.admm.bpdn'
D = np.random.randn(8, 16)
s = np.random.randn(8, 1)
lmbda = 0.1
## BPDN class
opt = bpdn.BPDN.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'bpdn_init.svg', **kwargs):
b = bpdn.BPDN(D, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'bpdn_solve.svg', **kwargs):
b.solve()
## BPDNJoint class
opt = bpdn.BPDNJoint.Options({'Verbose': False, 'MaxMainIter': 1})
mu = 0.01
with CallGraph(ct, mdnm, pth, 'bpdnjnt_init.svg', **kwargs):
b = bpdn.BPDNJoint(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'bpdnjnt_solve.svg', **kwargs):
b.solve()
## ElasticNet class
opt = bpdn.ElasticNet.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'elnet_init.svg', **kwargs):
b = bpdn.ElasticNet(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'elnet_solve.svg', **kwargs):
b.solve()
# BPDNProjL1 class
opt = bpdn.BPDNProjL1.Options({'Verbose': False, 'MaxMainIter': 1})
gamma = 2.0
with CallGraph(ct, mdnm, pth, 'bpdnprjl1_init.svg', **kwargs):
b = bpdn.BPDNProjL1(D, s, gamma, opt)
with CallGraph(ct, mdnm, pth, 'bpdnprjl1_solve.svg', **kwargs):
b.solve()
## MinL1InL2Ball class
opt = bpdn.MinL1InL2Ball.Options({'Verbose': False, 'MaxMainIter': 1})
epsilon = 1.0
with CallGraph(ct, mdnm, pth, 'bpdnml1l2_init.svg', **kwargs):
b = bpdn.MinL1InL2Ball(D, s, epsilon, opt)
with CallGraph(ct, mdnm, pth, 'bpdnml1l2_solve.svg', **kwargs):
b.solve()
#### cbpdn module
from sporco.admm import cbpdn
mdnm = 'sporco.admm.cbpdn'
D = np.random.randn(4, 4, 16)
s = np.random.randn(8, 8)
lmbda = 0.1
## ConvBPDN class
opt = cbpdn.ConvBPDN.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'cbpdn_init.svg', **kwargs):
b = cbpdn.ConvBPDN(D, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'cbpdn_solve.svg', **kwargs):
b.solve()
## ConvBPDNJoint class
opt = cbpdn.ConvBPDNJoint.Options({'Verbose': False, 'MaxMainIter': 1})
mu = 0.01
with CallGraph(ct, mdnm, pth, 'cbpdnjnt_init.svg', **kwargs):
b = cbpdn.ConvBPDNJoint(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnjnt_solve.svg', **kwargs):
b.solve()
## ConvElasticNet class
opt = cbpdn.ConvElasticNet.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'celnet_init.svg', **kwargs):
b = cbpdn.ConvElasticNet(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'celnet_solve.svg', **kwargs):
b.solve()
## ConvBPDNGradReg class
opt = cbpdn.ConvBPDNGradReg.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'cbpdngrd_init.svg', **kwargs):
b = cbpdn.ConvBPDNGradReg(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdngrd_solve.svg', **kwargs):
b.solve()
## ConvBPDNProjL1 class
opt = cbpdn.ConvBPDNProjL1.Options({'Verbose': False, 'MaxMainIter': 1})
gamma = 0.5
with CallGraph(ct, mdnm, pth, 'cbpdnprjl1_init.svg', **kwargs):
b = cbpdn.ConvBPDNProjL1(D, s, gamma, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnprjl1_solve.svg', **kwargs):
b.solve()
## ConvMinL1InL2Ball class
opt = cbpdn.ConvMinL1InL2Ball.Options({'Verbose': False, 'MaxMainIter': 1})
epsilon = 0.5
with CallGraph(ct, mdnm, pth, 'cbpdnml1l2_init.svg', **kwargs):
b = cbpdn.ConvMinL1InL2Ball(D, s, epsilon, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnml1l2_solve.svg', **kwargs):
b.solve()
## ConvBPDNMaskDcpl class
opt = cbpdn.ConvBPDNMaskDcpl.Options({'Verbose': False, 'MaxMainIter': 1})
W = np.ones(s.shape)
with CallGraph(ct, mdnm, pth, 'cbpdnmd_init.svg', **kwargs):
b = cbpdn.ConvBPDNMaskDcpl(D, s, lmbda, W, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnmd_solve.svg', **kwargs):
b.solve()
## ConvL1L1Grd class
opt = cbpdn.ConvL1L1Grd.Options({'Verbose': False, 'MaxMainIter': 1})
mu = 1e-2
with CallGraph(ct, mdnm, pth, 'cl1l1grd_init.svg', **kwargs):
b = cbpdn.ConvL1L1Grd(D, s, lmbda, mu, W, opt)
with CallGraph(ct, mdnm, pth, 'cl1l1grd_solve.svg', **kwargs):
b.solve()
#### cbpdntv module
from sporco.admm import cbpdntv
mdnm = 'sporco.admm.cbpdntv'
D = np.random.randn(4, 4, 16)
s = np.random.randn(8, 8)
lmbda = 0.1
mu = 0.01
## ConvBPDNScalarTV class
opt = cbpdntv.ConvBPDNScalarTV.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'cbpdnstv_init.svg', **kwargs):
b = cbpdntv.ConvBPDNScalarTV(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnstv_solve.svg', **kwargs):
b.solve()
## ConvBPDNVectorTV class
opt = cbpdntv.ConvBPDNVectorTV.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'cbpdnvtv_init.svg', **kwargs):
b = cbpdntv.ConvBPDNVectorTV(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnvtv_solve.svg', **kwargs):
b.solve()
## ConvBPDNRecTV class
opt = cbpdntv.ConvBPDNRecTV.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'cbpdnrtv_init.svg', **kwargs):
b = cbpdntv.ConvBPDNRecTV(D, s, lmbda, mu, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnrtv_solve.svg', **kwargs):
b.solve()
#### cbpdnin module
from sporco.admm import cbpdnin
mdnm = 'sporco.admm.cbpdnin'
D = np.random.randn(4, 4, 32)
s = np.random.randn(8, 8)
lmbda = 0.1
mu = 0.01
Wg = np.append(np.eye(16), np.eye(16), axis=-1)
## ConvBPDNInhib class
opt = cbpdnin.ConvBPDNInhib.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'cbpdnin_init.svg', **kwargs):
b = cbpdnin.ConvBPDNInhib(D,
s,
Wg,
Whn=4,
lmbda=lmbda,
mu=mu,
gamma=None,
opt=opt)
with CallGraph(ct, mdnm, pth, 'cbpdnin_solve.svg', **kwargs):
b.solve()
#### cmod module
from sporco.admm import cmod
mdnm = 'sporco.admm.cmod'
X = np.random.randn(8, 16)
S = np.random.randn(8, 16)
## CnstrMOD class
opt = cmod.CnstrMOD.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'cmod_init.svg', **kwargs):
b = cmod.CnstrMOD(X, S, opt=opt)
with CallGraph(ct, mdnm, pth, 'cmod_solve.svg', **kwargs):
b.solve()
#### ccmod module
from sporco.admm import ccmod
mdnm = 'sporco.admm.ccmod'
X = np.random.randn(8, 8, 1, 2, 1)
S = np.random.randn(8, 8, 2)
dsz = (4, 4, 1)
## ConvCnstrMOD_IterSM class
opt = ccmod.ConvCnstrMOD_IterSM.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'ccmodism_init.svg', **kwargs):
b = ccmod.ConvCnstrMOD_IterSM(X, S, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodism_solve.svg', **kwargs):
b.solve()
## ConvCnstrMOD_CG class
opt = ccmod.ConvCnstrMOD_CG.Options({
'Verbose': False,
'MaxMainIter': 1,
'CG': {
'MaxIter': 1
}
})
with CallGraph(ct, mdnm, pth, 'ccmodcg_init.svg', **kwargs):
b = ccmod.ConvCnstrMOD_CG(X, S, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodcg_solve.svg', **kwargs):
b.solve()
## ConvCnstrMOD_Consensus class
opt = ccmod.ConvCnstrMOD_Consensus.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'ccmodcnsns_init.svg', **kwargs):
b = ccmod.ConvCnstrMOD_Consensus(X, S, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodcnsns_solve.svg', **kwargs):
b.solve()
#### ccmodmd module
from sporco.admm import ccmodmd
mdnm = 'sporco.admm.ccmodmd'
X = np.random.randn(8, 8, 1, 2, 1)
S = np.random.randn(8, 8, 2)
W = np.array([1.0])
dsz = (4, 4, 1)
## ConvCnstrMODMaskDcpl_IterSM class
opt = ccmodmd.ConvCnstrMODMaskDcpl_IterSM.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'ccmodmdism_init.svg', **kwargs):
b = ccmodmd.ConvCnstrMODMaskDcpl_IterSM(X, S, W, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodmdism_solve.svg', **kwargs):
b.solve()
## ConvCnstrMODMaskDcpl_CG class
opt = ccmodmd.ConvCnstrMODMaskDcpl_CG.Options({
'Verbose': False,
'MaxMainIter': 1,
'CG': {
'MaxIter': 1
}
})
with CallGraph(ct, mdnm, pth, 'ccmodmdcg_init.svg', **kwargs):
b = ccmodmd.ConvCnstrMODMaskDcpl_CG(X, S, W, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodmdcg_solve.svg', **kwargs):
b.solve()
## ConvCnstrMODMaskDcpl_Consensus class
opt = ccmodmd.ConvCnstrMODMaskDcpl_Consensus.Options({
'Verbose': False,
'MaxMainIter': 1
})
with CallGraph(ct, mdnm, pth, 'ccmodmdcnsns_init.svg', **kwargs):
b = ccmodmd.ConvCnstrMODMaskDcpl_Consensus(X, S, W, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodmdcnsns_solve.svg', **kwargs):
b.solve()
#### tvl1 module
from sporco.admm import tvl1
mdnm = 'sporco.admm.tvl1'
s = np.random.randn(16, 16)
lmbda = 0.1
## TVL1Denoise class
opt = tvl1.TVL1Denoise.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'tvl1den_init.svg', **kwargs):
b = tvl1.TVL1Denoise(s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'tvl1den_solve.svg', **kwargs):
b.solve()
## TVL1Deconv class
opt = tvl1.TVL1Deconv.Options({'Verbose': False, 'MaxMainIter': 1})
h = np.random.randn(3, 3)
with CallGraph(ct, mdnm, pth, 'tvl1dcn_init.svg', **kwargs):
b = tvl1.TVL1Deconv(h, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'tvl1dcn_solve.svg', **kwargs):
b.solve()
#### tvl2 module
from sporco.admm import tvl2
mdnm = 'sporco.admm.tvl2'
s = np.random.randn(16, 16)
lmbda = 0.1
## TVL2Denoise class
opt = tvl2.TVL2Denoise.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'tvl2den_init.svg', **kwargs):
b = tvl2.TVL2Denoise(s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'tvl2den_solve.svg', **kwargs):
b.solve()
## TVL2Deconv class
opt = tvl2.TVL2Deconv.Options({'Verbose': False, 'MaxMainIter': 1})
h = np.random.randn(3, 3)
with CallGraph(ct, mdnm, pth, 'tvl2dcn_init.svg', **kwargs):
b = tvl2.TVL2Deconv(h, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'tvl2dcn_solve.svg', **kwargs):
b.solve()
srcmodflt = '^sporco.fista'
fnmsub = ('^sporco.fista.', '')
lnksub = (r'^([^\.]*).([^\.]*)(?:(.__init__|.__call__)|(.[^\.]*))',
lnkpfx + r'sporco.fista.\1.html#sporco.fista.\1.\2\4')
ct = jonga.CallTracer(srcmodflt=srcmodflt,
srcqnmflt=srcqnmflt,
dstqnmflt=dstqnmflt,
fnmsub=fnmsub,
grpflt=grpflt,
lnksub=lnksub)
#### fista.cbpdn module
from sporco.fista import cbpdn
mdnm = 'sporco.fista.cbpdn'
D = np.random.randn(4, 4, 16)
s = np.random.randn(8, 8)
lmbda = 0.1
## ConvBPDN class
opt = cbpdn.ConvBPDN.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'fista_cbpdn_init.svg', **kwargs):
b = cbpdn.ConvBPDN(D, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'fista_cbpdn_solve.svg', **kwargs):
b.solve()
#### fista.ccmod module
from sporco.fista import ccmod
mdnm = 'sporco.fista.ccmod'
X = np.random.randn(8, 8, 1, 2, 1)
S = np.random.randn(8, 8, 2)
dsz = (4, 4, 1)
## ConvCnstrMOD class
opt = ccmod.ConvCnstrMOD.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'ccmodfista_init.svg', **kwargs):
b = ccmod.ConvCnstrMOD(X, S, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodfista_solve.svg', **kwargs):
b.solve()
## ConvCnstrMODMask class
opt = ccmod.ConvCnstrMODMask.Options({'Verbose': False, 'MaxMainIter': 1})
with CallGraph(ct, mdnm, pth, 'ccmodmdfista_init.svg', **kwargs):
b = ccmod.ConvCnstrMODMask(X, S, W, dsz=dsz, opt=opt)
with CallGraph(ct, mdnm, pth, 'ccmodmdfista_solve.svg', **kwargs):
b.solve()
srcmodflt = '^sporco.dictlrn'
fnmsub = ('^sporco.dictlrn.', '')
lnksub = (r'^([^\.]*).([^\.]*)(?:(.__init__|.__call__)|(.[^\.]*))',
lnkpfx + r'sporco.dictlrn.\1.html#sporco.dictlrn.\1.\2\4')
ct = jonga.CallTracer(srcmodflt=srcmodflt,
srcqnmflt=srcqnmflt,
dstqnmflt=dstqnmflt,
fnmsub=fnmsub,
grpflt=grpflt,
lnksub=lnksub)
#### bpdndl module
from sporco.dictlrn import bpdndl
mdnm = 'sporco.dictlrn.bpdndl'
D0 = np.random.randn(8, 8)
S = np.random.randn(8, 16)
lmbda = 0.1
## BPDNDictLearn class
opt = bpdndl.BPDNDictLearn.Options({
'Verbose': False,
'MaxMainIter': 1,
'AccurateDFid': True
})
with CallGraph(ct, mdnm, pth, 'bpdndl_init.svg', **kwargs):
b = bpdndl.BPDNDictLearn(D0, S, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'bpdndl_solve.svg', **kwargs):
b.solve()
#### cbpdndl module
from sporco.dictlrn import cbpdndl
mdnm = 'sporco.dictlrn.cbpdndl'
D0 = np.random.randn(4, 4, 16)
s = np.random.randn(8, 8, 10)
lmbda = 0.1
## ConvBPDNDictLearn class
opt = cbpdndl.ConvBPDNDictLearn.Options({
'Verbose': False,
'MaxMainIter': 1,
'AccurateDFid': True
})
with CallGraph(ct, mdnm, pth, 'cbpdndl_init.svg', **kwargs):
b = cbpdndl.ConvBPDNDictLearn(D0, s, lmbda, opt)
with CallGraph(ct, mdnm, pth, 'cbpdndl_solve.svg', **kwargs):
b.solve()
#### cbpdndlmd module
from sporco.dictlrn import cbpdndlmd
mdnm = 'sporco.dictlrn.cbpdndlmd'
## ConvBPDNMaskDcplDictLearn class
W = np.array([1.0])
opt = cbpdndlmd.ConvBPDNMaskDictLearn.Options({
'Verbose': False,
'MaxMainIter': 1,
'AccurateDFid': True
})
with CallGraph(ct, mdnm, pth, 'cbpdnmddl_init.svg', **kwargs):
b = cbpdndlmd.ConvBPDNMaskDictLearn(D0, s, lmbda, W, opt)
with CallGraph(ct, mdnm, pth, 'cbpdnmddl_solve.svg', **kwargs):
b.solve()
