def test_zcs(): from utils import showArray from gaussian import gaussImage image = np.zeros((512,512), dtype=np.float32) image[255,255] = 1.0 smaller = gaussImage(image, 2) larger = gaussImage(image, 3) dog = smaller-larger zca = zcs(dog) showArray("smaller", smaller) showArray("larger", larger) showArray("dog", dog) showArray("zca", zca)
def test(): from median import median3x3 from gradient import gradient from hsi import rgb2hsi, hsi2rgb, joinChannels, splitChannels # Create a noisy image with an embedded white square image = np.zeros((201,199),dtype=np.float32) width,height = image.shape x,y = width/2, height/2 offset = 10 image[x-offset:x+offset,y-offset:y+offset] = 2 image += np.random.random_sample(image.shape) filtered = median3x3(image, 100) showArray("Noisy",image) showArray("Filtered",filtered) image = np.float32(imread("test.jpg")) image /= 256. showArray("Test HSI",image) r,g,b = splitChannels(image) h,s,i = rgb2hsi(r,g,b) showArray("I",i) showArray("S",s) showArray("H",h) from gaussian import gaussImage blur = gaussImage(i, 3) showArray("Blur", blur) blurmore = gaussImage(i,4) dog = blur-blurmore showArray("DOG", dog) g,a = gradient(i,5) showArray("Gradient",g) showArray("Angle", a) sat = np.ones_like(i) gimg = joinChannels(*hsi2rgb(a,sat,g)) showArray("Color gradient with angle", gimg) showArrayGrad("Grad angle", image, a) showArrayGrad("Grad vectors", image, a,g*10)