Replace import random by import random as rnd

Author: Belval

TextRecognitionDataGenerator/background_generator.py

@@ -1,7 +1,7 @@
 import cv2
 import math
 import os
-import random
+import random as rnd
 import numpy as np
 
 from PIL import Image, ImageDraw, ImageFilter
@@ -34,9 +34,9 @@ def quasicrystal(height, width):
     image = Image.new("L", (width, height))
     pixels = image.load()
 
-    frequency = random.random() * 30 + 20 # frequency
-    phase = random.random() * 2 * math.pi # phase
-    rotation_count = random.randint(10, 20) # of rotations
+    frequency = rnd.random() * 30 + 20 # frequency
+    phase = rnd.random() * 2 * math.pi # phase
+    rotation_count = rnd.randint(10, 20) # of rotations
 
     for kw in range(width):
         y = float(kw) / (width - 1) * 4 * math.pi - 2 * math.pi
@@ -59,23 +59,23 @@ def picture(height, width):
     pictures = os.listdir('./pictures')
 
     if len(pictures) > 0:
-        picture = Image.open('./pictures/' + pictures[random.randint(0, len(pictures) - 1)])
+        pic = Image.open('./pictures/' + pictures[rnd.randint(0, len(pictures) - 1)])
 
-        if picture.size[0] < width:
-            picture = picture.resize([width, int(picture.size[1] * (width / picture.size[0]))], Image.ANTIALIAS)
-        elif picture.size[1] < height:
-            picture.thumbnail([int(picture.size[0] * (height / picture.size[1])), height], Image.ANTIALIAS)
+        if pic.size[0] < width:
+            pic = pic.resize([width, int(pic.size[1] * (width / pic.size[0]))], Image.ANTIALIAS)
+        elif pic.size[1] < height:
+            pic.thumbnail([int(pic.size[0] * (height / pic.size[1])), height], Image.ANTIALIAS)
 
-        if (picture.size[0] == width):
+        if (pic.size[0] == width):
             x = 0
         else:
-            x = random.randint(0, picture.size[0] - width)
-        if (picture.size[1] == height):
+            x = rnd.randint(0, pic.size[0] - width)
+        if (pic.size[1] == height):
             y = 0
         else:
-            y = random.randint(0, picture.size[1] - height)
+            y = rnd.randint(0, pic.size[1] - height)
 
-        return picture.crop(
+        return pic.crop(
             (
                 x,
                 y,

TextRecognitionDataGenerator/computer_text_generator.py

@@ -1,4 +1,4 @@
-import random
+import random as rnd
 
 from PIL import Image, ImageColor, ImageFont, ImageDraw, ImageFilter
 
@@ -27,9 +27,9 @@ def _generate_horizontal_text(text, font, text_color, font_size, space_width, fi
     c1, c2 = colors[0], colors[-1]
 
     fill = (
-        random.randint(min(c1[0], c2[0]), max(c1[0], c2[0])),
-        random.randint(min(c1[1], c2[1]), max(c1[1], c2[1])),
-        random.randint(min(c1[2], c2[2]), max(c1[2], c2[2]))
+        rnd.randint(min(c1[0], c2[0]), max(c1[0], c2[0])),
+        rnd.randint(min(c1[1], c2[1]), max(c1[1], c2[1])),
+        rnd.randint(min(c1[2], c2[2]), max(c1[2], c2[2]))
     )
 
     for i, w in enumerate(words):
@@ -57,9 +57,9 @@ def _generate_vertical_text(text, font, text_color, font_size, space_width, fit)
     c1, c2 = colors[0], colors[-1]
 
     fill = (
-        random.randint(c1[0], c2[0]),
-        random.randint(c1[1], c2[1]),
-        random.randint(c1[2], c2[2])
+        rnd.randint(c1[0], c2[0]),
+        rnd.randint(c1[1], c2[1]),
+        rnd.randint(c1[2], c2[2])
     )
 
     for i, c in enumerate(text):

TextRecognitionDataGenerator/data_generator.py

@@ -1,5 +1,5 @@
 import os
-import random
+import random as rnd
 
 from PIL import Image, ImageFilter
 
@@ -39,7 +39,7 @@ def generate(cls, index, text, font, out_dir, size, extension, skewing_angle, ra
         else:
             image = computer_text_generator.generate(text, font, text_color, size, orientation, space_width, fit)
 
-        random_angle = random.randint(0-skewing_angle, skewing_angle)
+        random_angle = rnd.randint(0-skewing_angle, skewing_angle)
 
         rotated_img = image.rotate(skewing_angle if not random_skew else random_angle, expand=1)
 
@@ -117,7 +117,7 @@ def generate(cls, index, text, font, out_dir, size, extension, skewing_angle, ra
 
         final_image = background.filter(
             ImageFilter.GaussianBlur(
-                radius=(blur if not random_blur else random.randint(0, blur))
+                radius=(blur if not random_blur else rnd.randint(0, blur))
             )
         )
 

TextRecognitionDataGenerator/distorsion_generator.py

@@ -1,7 +1,7 @@
 import cv2
 import math
 import os
-import random
+import random as rnd
 import numpy as np
 
 from PIL import Image, ImageDraw, ImageFilter
@@ -78,4 +78,4 @@ def random(image, vertical=False, horizontal=False):
 
     max_offset = int(image.height ** 0.4)
 
-    return _apply_func_distorsion(image, vertical, horizontal, max_offset, (lambda x: random.randint(0, max_offset)))
+    return _apply_func_distorsion(image, vertical, horizontal, max_offset, (lambda x: rnd.randint(0, max_offset)))

TextRecognitionDataGenerator/handwritten_text_generator.py

@@ -1,7 +1,7 @@
 import os
 import pickle
 import numpy as np
-import random
+import random as rnd
 import tensorflow as tf
 import matplotlib.pyplot as plt
 import matplotlib.cm as cm
@@ -124,9 +124,9 @@ def generate(text, text_color):
         c1, c2 = colors[0], colors[-1]
 
         color = '#{:02x}{:02x}{:02x}'.format(
-            random.randint(min(c1[0], c2[0]), max(c1[0], c2[0])),
-            random.randint(min(c1[1], c2[1]), max(c1[1], c2[1])),
-            random.randint(min(c1[2], c2[2]), max(c1[2], c2[2]))
+            rnd.randint(min(c1[0], c2[0]), max(c1[0], c2[0])),
+            rnd.randint(min(c1[1], c2[1]), max(c1[1], c2[1])),
+            rnd.randint(min(c1[2], c2[2]), max(c1[2], c2[2]))
         )
 
         for word in text.split(' '):

TextRecognitionDataGenerator/run.py

@@ -1,6 +1,6 @@
 import argparse
 import os, errno
-import random
+import random as rnd
 import string
 import sys
 
@@ -336,7 +336,7 @@ def main():
         zip(
             [i for i in range(0, string_count)],
             strings,
-            [fonts[random.randrange(0, len(fonts))] for _ in range(0, string_count)],
+            [fonts[rnd.randrange(0, len(fonts))] for _ in range(0, string_count)],
             [args.output_dir] * string_count,
             [args.format] * string_count,
             [args.extension] * string_count,

TextRecognitionDataGenerator/string_generator.py

@@ -1,4 +1,4 @@
-import random
+import random as rnd
 import re
 import string
 import requests
@@ -33,8 +33,8 @@ def create_strings_from_dict(length, allow_variable, count, lang_dict):
     strings = []
     for _ in range(0, count):
         current_string = ""
-        for _ in range(0, random.randint(1, length) if allow_variable else length):
-            current_string += lang_dict[random.randrange(dict_len)][:-1]
+        for _ in range(0, rnd.randint(1, length) if allow_variable else length):
+            current_string += lang_dict[rnd.randrange(dict_len)][:-1]
             current_string += ' '
         strings.append(current_string[:-1])
     return strings
@@ -100,9 +100,9 @@ def create_strings_randomly(length, allow_variable, count, let, num, sym, lang):
     strings = []
     for _ in range(0, count):
         current_string = ""
-        for _ in range(0, random.randint(1, length) if allow_variable else length):
-            seq_len = random.randint(min_seq_len, max_seq_len)
-            current_string += ''.join([random.choice(pool) for _ in range(seq_len)])
+        for _ in range(0, rnd.randint(1, length) if allow_variable else length):
+            seq_len = rnd.randint(min_seq_len, max_seq_len)
+            current_string += ''.join([rnd.choice(pool) for _ in range(seq_len)])
             current_string += ' '
         strings.append(current_string[:-1])
     return strings