def round(self, decimals=0, out=None):
if decimals != 0:
raise NotImplementedError('only supports decimals=0')
s = arrayfire.round(self.d_array)
ret = ndarray(self.shape, dtype=pu.typemap(s.dtype()), af_array=s)
if(out):
out[:] = ret[:]
return ret
def round(self, decimals=0, out=None):
if decimals != 0:
raise NotImplementedError('only supports decimals=0')
s = arrayfire.round(self.d_array)
ret = ndarray(self.shape, dtype=pu.typemap(s.dtype()), af_array=s)
if (out):
out[:] = ret[:]
return ret
def simple_image(verbose=False):
display_func = _util.display_func(verbose)
print_func = _util.print_func(verbose)
a = 10 * af.randu(6, 6)
a3 = 10 * af.randu(5, 5, 3)
dx, dy = af.gradient(a)
display_func(dx)
display_func(dy)
display_func(af.resize(a, scale=0.5))
display_func(af.resize(a, odim0=8, odim1=8))
t = af.randu(3, 2)
display_func(af.transform(a, t))
display_func(af.rotate(a, 3.14))
display_func(af.translate(a, 1, 1))
display_func(af.scale(a, 1.2, 1.2, 7, 7))
display_func(af.skew(a, 0.02, 0.02))
h = af.histogram(a, 3)
display_func(h)
display_func(af.hist_equal(a, h))
display_func(af.dilate(a))
display_func(af.erode(a))
display_func(af.dilate3(a3))
display_func(af.erode3(a3))
display_func(af.bilateral(a, 1, 2))
display_func(af.mean_shift(a, 1, 2, 3))
display_func(af.medfilt(a))
display_func(af.minfilt(a))
display_func(af.maxfilt(a))
display_func(af.regions(af.round(a) > 3))
dx, dy = af.sobel_derivatives(a)
display_func(dx)
display_func(dy)
display_func(af.sobel_filter(a))
ac = af.gray2rgb(a)
display_func(ac)
display_func(af.rgb2gray(ac))
ah = af.rgb2hsv(ac)
display_func(ah)
display_func(af.hsv2rgb(ah))
display_func(af.color_space(a, af.CSPACE.RGB, af.CSPACE.GRAY))
def simple_image(verbose = False):
display_func = _util.display_func(verbose)
print_func = _util.print_func(verbose)
a = 10 * af.randu(6, 6)
a3 = 10 * af.randu(5,5,3)
dx,dy = af.gradient(a)
display_func(dx)
display_func(dy)
display_func(af.resize(a, scale=0.5))
display_func(af.resize(a, odim0=8, odim1=8))
t = af.randu(3,2)
display_func(af.transform(a, t))
display_func(af.rotate(a, 3.14))
display_func(af.translate(a, 1, 1))
display_func(af.scale(a, 1.2, 1.2, 7, 7))
display_func(af.skew(a, 0.02, 0.02))
h = af.histogram(a, 3)
display_func(h)
display_func(af.hist_equal(a, h))
display_func(af.dilate(a))
display_func(af.erode(a))
display_func(af.dilate3(a3))
display_func(af.erode3(a3))
display_func(af.bilateral(a, 1, 2))
display_func(af.mean_shift(a, 1, 2, 3))
display_func(af.medfilt(a))
display_func(af.minfilt(a))
display_func(af.maxfilt(a))
display_func(af.regions(af.round(a) > 3))
dx,dy = af.sobel_derivatives(a)
display_func(dx)
display_func(dy)
display_func(af.sobel_filter(a))
ac = af.gray2rgb(a)
display_func(ac)
display_func(af.rgb2gray(ac))
ah = af.rgb2hsv(ac)
display_func(ah)
display_func(af.hsv2rgb(ah))
display_func(af.color_space(a, af.CSPACE.RGB, af.CSPACE.GRAY))
af.display(af.hist_equal(a, h)) af.display(af.dilate(a)) af.display(af.erode(a)) af.display(af.dilate3(a3)) af.display(af.erode3(a3)) af.display(af.bilateral(a, 1, 2)) af.display(af.mean_shift(a, 1, 2, 3)) af.display(af.medfilt(a)) af.display(af.minfilt(a)) af.display(af.maxfilt(a)) af.display(af.regions(af.round(a) > 3)) dx,dy = af.sobel_derivatives(a) af.display(dx) af.display(dy) af.display(af.sobel_filter(a)) ac = af.gray2rgb(a) af.display(ac) af.display(af.rgb2gray(ac)) ah = af.rgb2hsv(ac) af.display(ah) af.display(af.hsv2rgb(ah)) af.display(af.color_space(a, af.AF_RGB, af.AF_GRAY))
def simple_arith(verbose = False):
display_func = _util.display_func(verbose)
print_func = _util.print_func(verbose)
a = af.randu(3,3,dtype=af.Dtype.u32)
b = af.constant(4, 3, 3, dtype=af.Dtype.u32)
display_func(a)
display_func(b)
c = a + b
d = a
d += b
display_func(c)
display_func(d)
display_func(a + 2)
display_func(3 + a)
c = a - b
d = a
d -= b
display_func(c)
display_func(d)
display_func(a - 2)
display_func(3 - a)
c = a * b
d = a
d *= b
display_func(c * 2)
display_func(3 * d)
display_func(a * 2)
display_func(3 * a)
c = a / b
d = a
d /= b
display_func(c / 2.0)
display_func(3.0 / d)
display_func(a / 2)
display_func(3 / a)
c = a % b
d = a
d %= b
display_func(c % 2.0)
display_func(3.0 % d)
display_func(a % 2)
display_func(3 % a)
c = a ** b
d = a
d **= b
display_func(c ** 2.0)
display_func(3.0 ** d)
display_func(a ** 2)
display_func(3 ** a)
display_func(a < b)
display_func(a < 0.5)
display_func(0.5 < a)
display_func(a <= b)
display_func(a <= 0.5)
display_func(0.5 <= a)
display_func(a > b)
display_func(a > 0.5)
display_func(0.5 > a)
display_func(a >= b)
display_func(a >= 0.5)
display_func(0.5 >= a)
display_func(a != b)
display_func(a != 0.5)
display_func(0.5 != a)
display_func(a == b)
display_func(a == 0.5)
display_func(0.5 == a)
display_func(a & b)
display_func(a & 2)
c = a
c &= 2
display_func(c)
display_func(a | b)
display_func(a | 2)
c = a
c |= 2
display_func(c)
display_func(a >> b)
display_func(a >> 2)
c = a
c >>= 2
display_func(c)
display_func(a << b)
display_func(a << 2)
c = a
c <<= 2
display_func(c)
display_func(-a)
display_func(+a)
display_func(~a)
display_func(a)
display_func(af.cast(a, af.Dtype.c32))
display_func(af.maxof(a,b))
display_func(af.minof(a,b))
display_func(af.rem(a,b))
a = af.randu(3,3) - 0.5
b = af.randu(3,3) - 0.5
display_func(af.abs(a))
display_func(af.arg(a))
display_func(af.sign(a))
display_func(af.round(a))
display_func(af.trunc(a))
display_func(af.floor(a))
display_func(af.ceil(a))
display_func(af.hypot(a, b))
display_func(af.sin(a))
display_func(af.cos(a))
display_func(af.tan(a))
display_func(af.asin(a))
display_func(af.acos(a))
display_func(af.atan(a))
display_func(af.atan2(a, b))
c = af.cplx(a)
d = af.cplx(a,b)
display_func(c)
display_func(d)
display_func(af.real(d))
display_func(af.imag(d))
display_func(af.conjg(d))
display_func(af.sinh(a))
display_func(af.cosh(a))
display_func(af.tanh(a))
display_func(af.asinh(a))
display_func(af.acosh(a))
display_func(af.atanh(a))
a = af.abs(a)
b = af.abs(b)
display_func(af.root(a, b))
display_func(af.pow(a, b))
display_func(af.pow2(a))
display_func(af.exp(a))
display_func(af.expm1(a))
display_func(af.erf(a))
display_func(af.erfc(a))
display_func(af.log(a))
display_func(af.log1p(a))
display_func(af.log10(a))
display_func(af.log2(a))
display_func(af.sqrt(a))
display_func(af.cbrt(a))
a = af.round(5 * af.randu(3,3) - 1)
b = af.round(5 * af.randu(3,3) - 1)
display_func(af.factorial(a))
display_func(af.tgamma(a))
display_func(af.lgamma(a))
display_func(af.iszero(a))
display_func(af.isinf(a/b))
display_func(af.isnan(a/a))
a = af.randu(5, 1)
b = af.randu(1, 5)
c = af.broadcast(lambda x,y: x+y, a, b)
display_func(a)
display_func(b)
display_func(c)
@af.broadcast
def test_add(aa, bb):
return aa + bb
display_func(test_add(a, b))
def simple_arith(verbose=False):
display_func = _util.display_func(verbose)
print_func = _util.print_func(verbose)
a = af.randu(3, 3)
b = af.constant(4, 3, 3)
display_func(a)
display_func(b)
c = a + b
d = a
d += b
display_func(c)
display_func(d)
display_func(a + 2)
display_func(3 + a)
c = a - b
d = a
d -= b
display_func(c)
display_func(d)
display_func(a - 2)
display_func(3 - a)
c = a * b
d = a
d *= b
display_func(c * 2)
display_func(3 * d)
display_func(a * 2)
display_func(3 * a)
c = a / b
d = a
d /= b
display_func(c / 2.0)
display_func(3.0 / d)
display_func(a / 2)
display_func(3 / a)
c = a % b
d = a
d %= b
display_func(c % 2.0)
display_func(3.0 % d)
display_func(a % 2)
display_func(3 % a)
c = a**b
d = a
d **= b
display_func(c**2.0)
display_func(3.0**d)
display_func(a**2)
display_func(3**a)
display_func(a < b)
display_func(a < 0.5)
display_func(0.5 < a)
display_func(a <= b)
display_func(a <= 0.5)
display_func(0.5 <= a)
display_func(a > b)
display_func(a > 0.5)
display_func(0.5 > a)
display_func(a >= b)
display_func(a >= 0.5)
display_func(0.5 >= a)
display_func(a != b)
display_func(a != 0.5)
display_func(0.5 != a)
display_func(a == b)
display_func(a == 0.5)
display_func(0.5 == a)
a = af.randu(3, 3, dtype=af.Dtype.u32)
b = af.constant(4, 3, 3, dtype=af.Dtype.u32)
display_func(a & b)
display_func(a & 2)
c = a
c &= 2
display_func(c)
display_func(a | b)
display_func(a | 2)
c = a
c |= 2
display_func(c)
display_func(a >> b)
display_func(a >> 2)
c = a
c >>= 2
display_func(c)
display_func(a << b)
display_func(a << 2)
c = a
c <<= 2
display_func(c)
display_func(-a)
display_func(+a)
display_func(~a)
display_func(a)
display_func(af.cast(a, af.Dtype.c32))
display_func(af.maxof(a, b))
display_func(af.minof(a, b))
display_func(af.rem(a, b))
a = af.randu(3, 3) - 0.5
b = af.randu(3, 3) - 0.5
display_func(af.abs(a))
display_func(af.arg(a))
display_func(af.sign(a))
display_func(af.round(a))
display_func(af.trunc(a))
display_func(af.floor(a))
display_func(af.ceil(a))
display_func(af.hypot(a, b))
display_func(af.sin(a))
display_func(af.cos(a))
display_func(af.tan(a))
display_func(af.asin(a))
display_func(af.acos(a))
display_func(af.atan(a))
display_func(af.atan2(a, b))
c = af.cplx(a)
d = af.cplx(a, b)
display_func(c)
display_func(d)
display_func(af.real(d))
display_func(af.imag(d))
display_func(af.conjg(d))
display_func(af.sinh(a))
display_func(af.cosh(a))
display_func(af.tanh(a))
display_func(af.asinh(a))
display_func(af.acosh(a))
display_func(af.atanh(a))
a = af.abs(a)
b = af.abs(b)
display_func(af.root(a, b))
display_func(af.pow(a, b))
display_func(af.pow2(a))
display_func(af.sigmoid(a))
display_func(af.exp(a))
display_func(af.expm1(a))
display_func(af.erf(a))
display_func(af.erfc(a))
display_func(af.log(a))
display_func(af.log1p(a))
display_func(af.log10(a))
display_func(af.log2(a))
display_func(af.sqrt(a))
display_func(af.cbrt(a))
a = af.round(5 * af.randu(3, 3) - 1)
b = af.round(5 * af.randu(3, 3) - 1)
display_func(af.factorial(a))
display_func(af.tgamma(a))
display_func(af.lgamma(a))
display_func(af.iszero(a))
display_func(af.isinf(a / b))
display_func(af.isnan(a / a))
a = af.randu(5, 1)
b = af.randu(1, 5)
c = af.broadcast(lambda x, y: x + y, a, b)
display_func(a)
display_func(b)
display_func(c)
@af.broadcast
def test_add(aa, bb):
return aa + bb
display_func(test_add(a, b))
def simple_image(verbose = False):
display_func = _util.display_func(verbose)
print_func = _util.print_func(verbose)
a = 10 * af.randu(6, 6)
a3 = 10 * af.randu(5,5,3)
dx,dy = af.gradient(a)
display_func(dx)
display_func(dy)
display_func(af.resize(a, scale=0.5))
display_func(af.resize(a, odim0=8, odim1=8))
t = af.randu(3,2)
display_func(af.transform(a, t))
display_func(af.rotate(a, 3.14))
display_func(af.translate(a, 1, 1))
display_func(af.scale(a, 1.2, 1.2, 7, 7))
display_func(af.skew(a, 0.02, 0.02))
h = af.histogram(a, 3)
display_func(h)
display_func(af.hist_equal(a, h))
display_func(af.dilate(a))
display_func(af.erode(a))
display_func(af.dilate3(a3))
display_func(af.erode3(a3))
display_func(af.bilateral(a, 1, 2))
display_func(af.mean_shift(a, 1, 2, 3))
display_func(af.medfilt(a))
display_func(af.minfilt(a))
display_func(af.maxfilt(a))
display_func(af.regions(af.round(a) > 3))
dx,dy = af.sobel_derivatives(a)
display_func(dx)
display_func(dy)
display_func(af.sobel_filter(a))
display_func(af.gaussian_kernel(3, 3))
display_func(af.gaussian_kernel(3, 3, 1, 1))
ac = af.gray2rgb(a)
display_func(ac)
display_func(af.rgb2gray(ac))
ah = af.rgb2hsv(ac)
display_func(ah)
display_func(af.hsv2rgb(ah))
display_func(af.color_space(a, af.CSPACE.RGB, af.CSPACE.GRAY))
a = af.randu(6,6)
b = af.unwrap(a, 2, 2, 2, 2)
c = af.wrap(b, 6, 6, 2, 2, 2, 2)
display_func(a)
display_func(b)
display_func(c)
display_func(af.sat(a))
a = af.randu(10,10,3)
display_func(af.rgb2ycbcr(a))
display_func(af.ycbcr2rgb(a))
a = af.randu(10, 10)
b = af.canny(a, low_threshold = 0.2, high_threshold = 0.8)
display_func(af.anisotropic_diffusion(a, 0.125, 1.0, 64, af.FLUX.QUADRATIC, af.DIFFUSION.GRAD))
af.display(+a) af.display(~a) af.display(a) af.display(af.cast(a, af.c32)) af.display(af.maxof(a, b)) af.display(af.minof(a, b)) af.display(af.rem(a, b)) a = af.randu(3, 3) - 0.5 b = af.randu(3, 3) - 0.5 af.display(af.abs(a)) af.display(af.arg(a)) af.display(af.sign(a)) af.display(af.round(a)) af.display(af.trunc(a)) af.display(af.floor(a)) af.display(af.ceil(a)) af.display(af.hypot(a, b)) af.display(af.sin(a)) af.display(af.cos(a)) af.display(af.tan(a)) af.display(af.asin(a)) af.display(af.acos(a)) af.display(af.atan(a)) af.display(af.atan2(a, b)) c = af.cplx(a) d = af.cplx(a, b) af.display(c)
def simple_image(verbose=False):
display_func = _util.display_func(verbose)
print_func = _util.print_func(verbose)
a = 10 * af.randu(6, 6)
a3 = 10 * af.randu(5, 5, 3)
dx, dy = af.gradient(a)
display_func(dx)
display_func(dy)
display_func(af.resize(a, scale=0.5))
display_func(af.resize(a, odim0=8, odim1=8))
t = af.randu(3, 2)
display_func(af.transform(a, t))
display_func(af.rotate(a, 3.14))
display_func(af.translate(a, 1, 1))
display_func(af.scale(a, 1.2, 1.2, 7, 7))
display_func(af.skew(a, 0.02, 0.02))
h = af.histogram(a, 3)
display_func(h)
display_func(af.hist_equal(a, h))
display_func(af.dilate(a))
display_func(af.erode(a))
display_func(af.dilate3(a3))
display_func(af.erode3(a3))
display_func(af.bilateral(a, 1, 2))
display_func(af.mean_shift(a, 1, 2, 3))
display_func(af.medfilt(a))
display_func(af.minfilt(a))
display_func(af.maxfilt(a))
display_func(af.regions(af.round(a) > 3))
dx, dy = af.sobel_derivatives(a)
display_func(dx)
display_func(dy)
display_func(af.sobel_filter(a))
display_func(af.gaussian_kernel(3, 3))
display_func(af.gaussian_kernel(3, 3, 1, 1))
ac = af.gray2rgb(a)
display_func(ac)
display_func(af.rgb2gray(ac))
ah = af.rgb2hsv(ac)
display_func(ah)
display_func(af.hsv2rgb(ah))
display_func(af.color_space(a, af.CSPACE.RGB, af.CSPACE.GRAY))
a = af.randu(6, 6)
b = af.unwrap(a, 2, 2, 2, 2)
c = af.wrap(b, 6, 6, 2, 2, 2, 2)
display_func(a)
display_func(b)
display_func(c)
display_func(af.sat(a))
a = af.randu(10, 10, 3)
display_func(af.rgb2ycbcr(a))
display_func(af.ycbcr2rgb(a))
a = af.randu(10, 10)
b = af.canny(a, low_threshold=0.2, high_threshold=0.8)
display_func(
af.anisotropic_diffusion(a, 0.125, 1.0, 64, af.FLUX.QUADRATIC,
af.DIFFUSION.GRAD))
af.display(af.hist_equal(a, h)) af.display(af.dilate(a)) af.display(af.erode(a)) af.display(af.dilate3(a3)) af.display(af.erode3(a3)) af.display(af.bilateral(a, 1, 2)) af.display(af.mean_shift(a, 1, 2, 3)) af.display(af.medfilt(a)) af.display(af.minfilt(a)) af.display(af.maxfilt(a)) af.display(af.regions(af.round(a) > 3)) dx, dy = af.sobel_derivatives(a) af.display(dx) af.display(dy) af.display(af.sobel_filter(a)) ac = af.gray2rgb(a) af.display(ac) af.display(af.rgb2gray(ac)) ah = af.rgb2hsv(ac) af.display(ah) af.display(af.hsv2rgb(ah)) af.display(af.color_space(a, af.AF_RGB, af.AF_GRAY))
def unwrap_dct(arr):
return arr + af.round(
(laplacian(af.cos(arr) * laplacian(af.sin(arr)) -
af.sin(arr) * laplacian(af.cos(arr)),
inverse=True) - arr) / 2 / np.pi) * 2 * np.pi
af.display(+a) af.display(~a) af.display(a) af.display(af.cast(a, af.c32)) af.display(af.maxof(a,b)) af.display(af.minof(a,b)) af.display(af.rem(a,b)) a = af.randu(3,3) - 0.5 b = af.randu(3,3) - 0.5 af.display(af.abs(a)) af.display(af.arg(a)) af.display(af.sign(a)) af.display(af.round(a)) af.display(af.trunc(a)) af.display(af.floor(a)) af.display(af.ceil(a)) af.display(af.hypot(a, b)) af.display(af.sin(a)) af.display(af.cos(a)) af.display(af.tan(a)) af.display(af.asin(a)) af.display(af.acos(a)) af.display(af.atan(a)) af.display(af.atan2(a, b)) c = af.cplx(a) d = af.cplx(a,b) af.display(c)
def simple_image(verbose=False):
display_func = _util.display_func(verbose)
a = 10 * af.randu(6, 6)
a3 = 10 * af.randu(5, 5, 3)
dx, dy = af.gradient(a)
display_func(dx)
display_func(dy)
display_func(af.resize(a, scale=0.5))
display_func(af.resize(a, odim0=8, odim1=8))
t = af.randu(3, 2)
display_func(af.transform(a, t))
display_func(af.rotate(a, 3.14))
display_func(af.translate(a, 1, 1))
display_func(af.scale(a, 1.2, 1.2, 7, 7))
display_func(af.skew(a, 0.02, 0.02))
h = af.histogram(a, 3)
display_func(h)
display_func(af.hist_equal(a, h))
display_func(af.dilate(a))
display_func(af.erode(a))
display_func(af.dilate3(a3))
display_func(af.erode3(a3))
display_func(af.bilateral(a, 1, 2))
display_func(af.mean_shift(a, 1, 2, 3))
display_func(af.medfilt(a))
display_func(af.minfilt(a))
display_func(af.maxfilt(a))
display_func(af.regions(af.round(a) > 3))
display_func(
af.confidenceCC(af.randu(10,
10), (af.randu(2) * 9).as_type(af.Dtype.u32),
(af.randu(2) * 9).as_type(af.Dtype.u32), 3, 3, 10,
0.1))
dx, dy = af.sobel_derivatives(a)
display_func(dx)
display_func(dy)
display_func(af.sobel_filter(a))
display_func(af.gaussian_kernel(3, 3))
display_func(af.gaussian_kernel(3, 3, 1, 1))
ac = af.gray2rgb(a)
display_func(ac)
display_func(af.rgb2gray(ac))
ah = af.rgb2hsv(ac)
display_func(ah)
display_func(af.hsv2rgb(ah))
display_func(af.color_space(a, af.CSPACE.RGB, af.CSPACE.GRAY))
a = af.randu(6, 6)
b = af.unwrap(a, 2, 2, 2, 2)
c = af.wrap(b, 6, 6, 2, 2, 2, 2)
display_func(a)
display_func(b)
display_func(c)
display_func(af.sat(a))
a = af.randu(10, 10, 3)
display_func(af.rgb2ycbcr(a))
display_func(af.ycbcr2rgb(a))
a = af.randu(10, 10)
b = af.canny(a, low_threshold=0.2, high_threshold=0.8)
display_func(
af.anisotropic_diffusion(a, 0.125, 1.0, 64, af.FLUX.QUADRATIC,
af.DIFFUSION.GRAD))
a = af.randu(10, 10)
psf = af.gaussian_kernel(3, 3)
cimg = af.convolve(a, psf)
display_func(
af.iterativeDeconv(cimg, psf, 100, 0.5, af.ITERATIVE_DECONV.LANDWEBER))
display_func(
af.iterativeDeconv(cimg, psf, 100, 0.5,
af.ITERATIVE_DECONV.RICHARDSONLUCY))
display_func(af.inverseDeconv(cimg, psf, 1.0, af.INVERSE_DECONV.TIKHONOV))
