import math
import numpy as np
import matplotlib.pyplot as plt
from skimage import io, segmentation, color
from skimage.future import graph
%matplotlib inline
image = io.imread("../images/head-ct.jpg")
rows, cols, depth = image.shape
print(rows)
print(cols)
print(depth)
fig, ax = plt.subplots(figsize=(8, 8))
plt.imshow(image)
segments_slic = segmentation.slic(image, n_segments=1200, compactness=30, sigma=.5)
print(np.amax(segments_slic))
fig, ax = plt.subplots(figsize=(8, 8))
plt.imshow(segmentation.mark_boundaries(image, segments_slic))
plt.show()
g = graph.rag_mean_color(image, segments_slic, mode='similarity')
labels = graph.cut_normalized(segments_slic, g)
output = color.label2rgb(labels, image, kind='avg')
fig, ax = plt.subplots(figsize=(8, 8))
plt.imshow(output)
plt.show()