class SparsityDecoratorFG(FilterGenerator):
def __init__(self, filterGenerator, sparseProb, seed=None):
self._fg = filterGenerator
self._numGen = NumberGenerator(0, 1, seed)
self._prob = sparseProb
def next(self):
filt = self._fg.next()
height, width = filt.shape
for h in xrange(height):
for w in xrange(width):
if self._numGen.getNumber(0, 1) < self._prob:
filt[h][w] = 0
return filt
class SparsityDecoratorFG(FilterGenerator):
def __init__(self, filterGenerator, sparseProb, seed=None):
self._fg = filterGenerator
self._numGen = NumberGenerator(0, 1, seed)
self._prob = sparseProb
def next(self):
filt = self._fg.next()
height, width = filt.shape
for h in xrange(height):
for w in xrange(width):
if self._numGen.getNumber(0, 1) < self._prob:
filt[h][w] = 0
return filt
