Create imageProcessing_preCNN.py

imageProcessing_preCNN.py

@@ -0,0 +1,245 @@
+# The Captcha image consists of four characters that are either a string of letters or a combination of letters with numbers. In some of the combinations, the captcha characters contain all letters but they are not aligned cleanly. Some of the letters are joined together or diagonally aligned, therefore correctly separating them in order to
+# make them easily recognised as letters is one of the tasks at hand. In the instances of Captcha characters that contain both numbers and letters, the position of the number is not guaranteed and the letters are slanted.
+# The expected outcome is to correctly identify each of the Captcha characters from the image. We are doing this inorder to measure the efficiency of the neural networks to correctly identify the Captcha characters. The label variable is the letter or number from
+# the image of the separated Captcha characters.
+
+# Ways to Instal OPENCV (choose one that works)
+
+!pip install opencv
+conda install -c conda-forge/label/gcc7 opencv
+conda install -c menpo opencv
+
+# Common imports
+import numpy as np
+import os, os.path
+import cv2
+import imutils
+from imutils import paths
+import glob
+import pandas as pd
+from PIL import Image
+import zipfile
+with zipfile.ZipFile("archive.zip","r") as zip_ref:
+    zip_ref.extractall("targetdir")
+import matplotlib as mpl
+import matplotlib.pyplot as plt
+mpl.rc('axes', labelsize=14)
+mpl.rc('xtick', labelsize=12)
+mpl.rc('ytick', labelsize=12)
+
+
+# Set Random Seed
+np.random.seed(42)
+
+# Get File Directory
+os.getcwd()
+
+# Reading in images from directory saved in path object
+# define read image function for reading images
+def read_image(image_file_path):
+    """Read in an image file."""
+    bgr_img = cv2.imread(image_file_path)
+    b,g,r = cv2.split(bgr_img)       # get b,g,r
+    rgb_img = cv2.merge([r,g,b])     # switch it to rgb
+
+    return rgb_img
+
+#Define image directory in image_file_Path
+image_file_Path ="---"
+images = []
+labels = []
+
+# read images and labels
+for image_file_path in imutils.paths.list_images(image_file_Path):
+    image_file = read_image(image_file_path)
+    label = image_file_path.split('/')[7]
+    images.append(image_file)
+    labels.append(label)
+
+# Updating label names
+newLabels = []
+
+for label in labels:
+    labelHere = label.split('.')[0]
+    newLabels.append(labelHere)
+
+# Coverting ptyhon lists into NumPy Arrays
+images = np.array(images)
+labels = np.array(newLabels)
+
+# A dataset of 9,955 of unique CAPTCHA images each with its label as the filename was used for this research. 
+# However, machine learning classification requires a one-to-many relationship between a label and in this context the CAPTCHA images. 
+# Therefore, uniqueness of the CAPTCHA images is problematic for a machine learning process.
+
+# Breaking up CAPTCHA character images into individual character images
+
+# This how the images look like rn CAPTCHA character images
+some_digit = images[300]
+plt.imshow(some_digit, cmap = mpl.cm.binary,
+           interpolation="nearest")
+plt.axis("off")
+plt.show()
+labels[300]  #image's label
+
+# Define Padding and Threshold functions
+def pureBlackWhiteConversionThreshold(image):
+    # Add some extra padding around the image
+    imagePadded = cv2.copyMakeBorder(image, 8, 8, 8, 8, cv2.BORDER_REPLICATE)
+    gray = cv2.cvtColor(imagePadded, cv2.COLOR_RGB2GRAY)
+    # threshold the image (convert it to pure black and white)
+    imagethresholded = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV | cv2.THRESH_OTSU)[1]    
+    return imagethresholded
+
+def pureBlackWhiteConversionOGImage(image):
+    # Add some extra padding around the image
+    imagePadded = cv2.copyMakeBorder(image, 8, 8, 8, 8, cv2.BORDER_REPLICATE)
+    gray = cv2.cvtColor(imagePadded, cv2.COLOR_RGB2GRAY)
+
+    return gray
+
+# This is a Padded and Thresholded image
+padded_ThreshImage300 = pureBlackWhiteConversionThreshold(images[300])
+some_digit = padded_ThreshImage300
+plt.imshow(some_digit, cmap = mpl.cm.binary,
+           interpolation="nearest")
+plt.axis("off")
+plt.show()
+
+# Define function for deteriming the regions for individual character images in each captcha image
+def regionsOfLetters(image):
+
+     # find the contours (continuous blobs of pixels) the image
+    contours = cv2.findContours(image, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+
+    # Hack for compatibility with different OpenCV versions
+    contours = contours[0] if imutils.is_cv2() else contours[1]
+
+    letter_image_regions = []
+
+    # Now we can loop through each of the four contours and extract the letter
+    # inside of each one
+    for contour in contours:
+        # Get the rectangle that contains the contour
+        (x, y, w, h) = cv2.boundingRect(contour)
+
+        # Compare the width and height of the contour to detect letters that
+        # are conjoined into one chunk
+        if w / h > 1.25:
+            # This contour is too wide to be a single letter!
+            # Split it in half into two letter regions!
+            half_width = int(w / 2)
+            letter_image_regions.append((x, y, half_width, h))
+            letter_image_regions.append((x + half_width, y, half_width, h))
+        else:
+            # This is a normal letter by itself
+            letter_image_regions.append((x, y, w, h))
+    # If we found more or less than 4 letters in the captcha, our letter extraction
+    # didn't work correcly. Skip the image instead of saving bad training data!
+    #if len(letter_image_regions) != 4:
+       # continue
+
+    # Sort the detected letter images based on the x coordinate to make sure
+    # we are processing them from left-to-right so we match the right image
+    # with the right letter
+    letter_image_regions = sorted(letter_image_regions, key=lambda x: x[0])
+
+    return letter_image_regions
+
+# Function for Cutting out the indivual characters from the captcha imaes
+def extractLetters(letter_image_regions, image):
+    # Save out each letter as a single image
+    letter_images =[]
+    for letter_bounding_box in letter_image_regions:
+        # Grab the coordinates of the letter in the image
+        x, y, w, h = letter_bounding_box
+
+        # Extract the letter from the original image with a 2-pixel margin around the edge
+        letter_image = image[y - 2:y + h + 2, x - 2:x + w + 2]
+        #image_file1 = read_image(letter_image)
+        letter_images.append(letter_image)
+    return letter_images
+
+# function for resizing the cut-out images 
+def expand2square(image):
+    desired_size = 28
+    im = image
+    old_size = im.shape[:2] # old_size is in (height, width) format
+
+    ratio = float(desired_size)/max(old_size)
+    new_size = tuple([int(x*ratio) for x in old_size])
+
+    # new_size should be in (width, height) format
+
+    im = cv2.resize(im, (new_size[1], new_size[0]))
+
+    delta_w = desired_size - new_size[1]
+    delta_h = desired_size - new_size[0]
+    top, bottom = delta_h//2, delta_h-(delta_h//2)
+    left, right = delta_w//2, delta_w-(delta_w//2)
+
+    color = [255, 255, 255]
+    new_im = cv2.copyMakeBorder(im, top, bottom, left, right, cv2.BORDER_CONSTANT,
+        value=color)
+    return new_im
+
+# The cutting out Process
+letterImages = []
+letterImageLabels = []
+
+
+# loop over the unseparated image list
+for label, image in zip(labels, images):
+
+    padded_ThreshImage300 = pureBlackWhiteConversionThreshold(image)
+    letter_image_regions = regionsOfLetters(padded_ThreshImage300)
+    grayScaleImage = pureBlackWhiteConversionOGImage(image)
+    letter_image_List = extractLetters(letter_image_regions,grayScaleImage)
+     #image_reshape = letter_bounding_box.reshape(L0, L1)
+    for letter_bounding_box, letter_text in zip(letter_image_List, label): 
+        L0=letter_bounding_box.shape[0]
+        L1=letter_bounding_box.shape[1]
+
+        imageResize = expand2square(letter_bounding_box)
+        letterImages.append(imageResize)
+        letterImageLabels.append(letter_text)
+
+# Display an example
+def plot_digit(image):
+    some_digit = image
+    plt.imshow(some_digit, cmap = mpl.cm.binary,
+               interpolation="nearest")
+    plt.axis("off")
+    plt.show()
+
+def plot_digits(instances, images_per_row=10, **options):
+    size = 28
+    images_per_row = min(len(instances), images_per_row)
+    images = [instance.reshape(size,size) for instance in instances]
+    n_rows = (len(instances) - 1) // images_per_row + 1
+    row_images = []
+    n_empty = n_rows * images_per_row - len(instances)
+    images.append(np.zeros((size, size * n_empty)))
+    for row in range(n_rows):
+        rimages = images[row * images_per_row : (row + 1) * images_per_row]
+        row_images.append(np.concatenate(rimages, axis=1))
+    image = np.concatenate(row_images, axis=0)
+    plt.imshow(image, cmap = mpl.cm.binary, **options)
+    plt.axis("off")
+
+index, = np.where(labels == 'F')
+index
+plt.figure(figsize=(15, 15))
+example_images = np.r_[images[[14,39,51,39702,39752]], 
+                       images[[7,56,61,39703,39714]],
+                       images[[45,   198,   352,39705, 39719]], 
+                       images[[2,    12,    52, 39698, 39712]], 
+                       images[[3,    26,    87, 39612, 39619]]]
+
+
+plot_digits(example_images, images_per_row=5)
+
+#To deal with the uniqueness problem of the dataset, the solution was to separate  the CAPTCHA images into the individual 4 characters that make up the CHAPTCHA image. 
+# This was to make each character into its own image. The resulting dataset has 39,754 images with one character per image. 
+# The new dataset satisfies the one-to-many relationship between the images and 
+# the following 32 characters labels {'2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 
+# 'H', 'J', 'K', 'L', 'M', 'N', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z'}