main.py

from PIL import Image, ImageDraw, ImageFont
from random import randint, random
from string import ascii_uppercase, ascii_lowercase, digits
import numpy as np
import cv2
import imutils
from matplotlib.pyplot import imshow
import matplotlib.pyplot as plt
import os

# Reference: https://thispointer.com/python-how-to-get-list-of-files-in-directory-and-sub-directories/
# For the given path, get the List of all files in the directory tree


def getListOfFiles(dirName):
    # create a list of file and sub directories
    # names in the given directory
    listOfFile = os.listdir(dirName)
    allFiles = list()
    # Iterate over all the entries
    for entry in listOfFile:
        # Create full path
        fullPath = os.path.join(dirName, entry)
        # If entry is a directory then get the list of files in this directory
        if os.path.isdir(fullPath):
            allFiles = allFiles + getListOfFiles(fullPath)
        else:
            allFiles.append(fullPath)

    return allFiles


os.system("mkdir -p data")
all_char = ascii_uppercase + digits
font_paths = []
for c in getListOfFiles("./fonts/"):
    if c[len(c)-4:] == ".ttf":
        font_paths.append(c)
myfonts = [ImageFont.truetype(font=i, size=250) for i in font_paths]

f_ind = 1
for char in all_char[:]:
    freq = 0
    for ind in range(100):
        freq += 1
        out = Image.new("RGB", (400, 400), (255, 255, 255))
        font = myfonts[randint(0, len(myfonts)-1)]
        # print(font.getname()[0])
        d = ImageDraw.Draw(out)
        d.text((75, 40), char, font=font, fill=(0, 0, 0))
        out = out.rotate(10*random()-5)
        out.save("_.png")
        img = cv2.imread('_.png')
        img = img[20:380, 20:380]
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        _, thresh = cv2.threshold(gray, 240, 255, cv2.THRESH_BINARY)
        cv2.bitwise_not(thresh, thresh)
        cnts = cv2.findContours(
            thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
        cnts = cnts[0]
        c = max(cnts, key=cv2.contourArea)
        left = tuple(c[c[:, :, 0].argmin()][0])
        right = tuple(c[c[:, :, 0].argmax()][0])
        distance = np.sqrt((right[0] - left[0])**2 + (right[1] - left[1])**2)
        x, y, w, h = cv2.boundingRect(c)
        centx = np.sqrt(((right[0] + left[0])**2)/4)
        centy = np.sqrt(((right[1] + left[1])**2)/4)

        # print(centx, centy)
        # print(x, y)
        # print(w, h)

        # cv2.rectangle(img, (x, y), (x+w, y+h), (0, 255, 0), 2)
        cropped = img[y:y+h, x:x+w]
        b = 15
        cropped = cv2.copyMakeBorder(
            cropped, 
            b, b, b, b, 
            cv2.BORDER_CONSTANT, value=[255, 255, 255])
        resized = cv2.resize(cropped, (40, 40))
        # imshow(resized), plt.show()

        imagem = cv2.bitwise_not(resized)

        os.system(str("mkdir -p ./data/"+"Sample"+str(f_ind).zfill(3)))
        cv2.imwrite("./data/"+"Sample"+str(f_ind).zfill(3) +
                    "/"+char+"_"+str(ind)+"_"+font.getname()[0]+".png", imagem)
    print(char, ":", freq)
    f_ind += 1