histograms

import matplotlib
matplotlib.use('TkAgg')
import matplotlib.pyplot as plt
import numpy as np
import matplotlib.image as mpimg

"""Takes binary images as inputs. Calculates the frequency histograms (both horizontal and vertical)
Should be used after cutting away the major areas of noise (use on smaller images)"""

image = "Binary.png" # your input image should be white text on black background

binary = mpimg.imread(image) # collection of lists

fig = plt.figure()

a = fig.add_subplot(1, 2, 1)

print binary
print binary.shape # y, x (in dims)

# want to compute the sum along the y-dim
sum1 = np.sum(binary, axis=1)

plt.title("Horizontal Frequencies")

plt.plot(sum1) # this gives the histogram for horizontal direction (->), with y varying

a = fig.add_subplot(1, 2, 2)
plt.title("Vertical Frequencies")
sum2 = np.sum(binary, axis=0) # get the histogram for ^ dim, vertical, with x varying

plt.plot(sum2)

plt.show()