Crop classes afgemaakt en test toegevoegd

src/crop.py

 from PIL import Image
-from Pylab import *
-from LBP import domain_iterator
+from pylab import *
 
-THRESHOLD = 0.5
+class LetterCropper:
 
-im = Image.open('../.jpg')
-im = Image.convert('L', im)
+    THRESHOLD = 0.5
 
-outer_bounds = get_outer_bounds()
+    def __init__(self, image_path):
+        self.set_image(image_path)
         
-im.crop(outer_bounds)
+    def set_image(self, image_path):
+        self.source_image = imread(image_path)
                 
-imshow(im)
+    def get_cropped_letter(self):
+        self.convert_image_to_grayscale()
+        self.determine_letter_bounds()
+        self.crop_image()
+        return self.cropped_letter
 
-show()
+    def convert_image_to_grayscale(self):
+        self.cropped_letter = self.source_image.sum(axis=2) / 3
 
-def get_outer_bound():
-    min_x = len(im[0])
+    def determine_letter_bounds(self):
+        image_width = len(self.cropped_letter[0])
+        image_height = len(self.cropped_letter)
+    
+        min_x = image_width
         max_x = 0
-    min_y = len(im)
+        min_y = image_height
         max_y = 0
 
-    for y in xrange(len(im)):
-        for x in xrange(len(im[0])):
-            if im[y, x] > THRESHOLD:
+        for y in xrange(image_height):
+            for x in xrange(image_width):
+                if self.cropped_letter[y, x] < self.THRESHOLD:
                     if x < min_x: min_x = x
                     if y < min_y: min_y = y
                     if x > max_x: max_x = x
                     if y > max_y: max_y = y
-    return (min_x, min_y, max_x, max_y)
-            
-        
         
+        self.letter_bounds = (min_x, min_y, max_x, max_y)
         
+    def crop_image(self):
+        self.cropped_letter = self.cropped_letter[self.letter_bounds[1] : self.letter_bounds[3], 
+                                                  self.letter_bounds[0] : self.letter_bounds[2]]
 

src/crop_test.py

+from pylab import *
+from crop import LetterCropper
+
+letter_cropper = LetterCropper("../images/test.png")
+cropped_letter = letter_cropper.get_cropped_letter()
+
+imshow(cropped_letter, cmap="gray")
+
+show()
+
+