def color_channel_entropy_test(file):
image = cv2.imread(file, cv2.IMREAD_COLOR)
image = cv2.GaussianBlur(image,(3,3),0)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
blues = bv.extract_blue(image)
greens = bv.extract_green(image)
reds = bv.extract_red(image)
blueMap = np.copy(image)
redMap = np.copy(image)
greenMap = np.copy(image)
grayMap = np.copy(image)
bui.show_images([blues, greens, reds])
fsize = 50
step = 10
fsize = 5
step = 2
rsize = step
print("eGray: %f eBlue: %f, eGreen: %f, eRed: %f"%(bv.entropy(gray), bv.entropy(blues), bv.entropy(greens), bv.entropy(reds)))
emap = bv.entropy_map(gray, fsize, step)
bmap = bv.entropy_map(blues, fsize, step)
gmap = bv.entropy_map(greens, fsize, step)
rmap = bv.entropy_map(reds, fsize, step)
bui.paint_map(grayMap, bmap, fsize, step, rsize)
bui.paint_map(blueMap, bmap, fsize, step, rsize)
bui.paint_map(greenMap, gmap, fsize, step, rsize)
bui.paint_map(redMap, rmap, fsize, step, rsize)
bui.show_images([image, gray, grayMap, reds, greens, blues, redMap, greenMap, blueMap])
def partial_entropy_test(file):
image = cv2.imread(file, 0)
height, width = image.shape[:2]
colored_image = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
ps = 3
sizeY = int(height / ps)
sizeX = int(width / ps)
fsize = 50
step = 10
fsize = 10
step = 5
rsize = step
images = [image]
for y in range(0, ps):
for x in range(0, ps):
y1 = y*sizeY
y2 = y1+sizeY
x1 = x*sizeX
x2 = x1 + sizeX
nimage = image[y1:y2, x1:x2]
emap = bv.entropy_map(nimage, fsize, step)
cimage = cv2.cvtColor(nimage, cv2.COLOR_GRAY2RGB)
bui.paint_map(cimage, emap, fsize, step, rsize)
images.append(nimage)
images.append(cimage)
bui.show_images(images)
def color_extraction_test(file): image = cv2.imread(file, cv2.IMREAD_COLOR) # print(image.shape) # print(np.multiply(image, (1, 1, 1)).shape) # image = map(lambda x: x * (1, 1, 1), image) #image = np.arange(5*5*3).reshape(5,5,3) #image = np.ones([5, 5, 3]) * 255 bui.show_images([image, bv.extract_blue(image), bv.extract_green(image), bv.extract_red(image)])
def entropy_map_test(file): image = cv2.imread(file, 0) colored_image = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB) bui.show_images([image]) fsize = 50 step = 10 fsize = 10 step = 5 rsize = step emap = bv.entropy_map(image, fsize, step) bui.paint_map(colored_image, emap, fsize, step, rsize) bui.show_images([image, colored_image])
def symmetry_strip_test(file): image = cv2.imread(file, 0) vsym = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB) hsym = np.copy(vsym) tsym = np.copy(vsym) size = 10 smap = bv.symmetry_strip_map(image, size) vmap = bv.symmetry_strip_axis_map(image, 1, size) hmap = bv.symmetry_strip_axis_map(image, 0, size) bui.paint_symmetry(tsym, smap, size) bui.paint_axis_symmetry(vsym, vmap, size) bui.paint_axis_symmetry(hsym, hmap, size) bui.show_images([image, vsym, hsym, tsym])
def entropy_focus_test(file):
image = cv2.imread(file, 0)
image = cv2.GaussianBlur(image,(9,9),0)
bui.show_images([image])
images = [image]
for f in [[80, 20], [50, 10], [10, 5], [5, 2]]:
fsize = f[0]
step = f[1]
rsize = step
emap = bv.entropy_map(image, fsize, step)
cimage = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
bui.paint_map(cimage, emap, fsize, step, rsize)
images.append(cimage)
bui.show_images(images)