draw rectangles ("people") using Augmentor

add_people.py

import Augmentor
from  Augmentor.Operations import Operation
class GeneratePeopleShapes(Augmentor.Operations.Operation):   
    def __init__(self, probability):

        Operation.__init__(self, probability)

    @staticmethod
    def gen_random_rectangle_coords(top, bottom, left, right):
        width =random.randint(5, 59)
        height = random.randint(30,60)
        x_center = random.randint(left, right)
        y_center = random.randint(top, bottom)

        tl = (x_center-width, y_center+height)
        br = (x_center+width, y_center-height)
        return tl, br


    def perform_operation(self, images):
        
        def do(image):
            arr = np.array(image)
            tl, br = self.gen_random_rectangle_coords(10, 30, 10, 150)
            color = tuple(random.choice(range(0, 100)) for i in range(3))
            arr = cv2.rectangle(arr, tl, br, color,-1)
            image = Image.fromarray(arr)
            return image
        
        augmented_images = []
        
        for image in images:
            augmented_images.append(do(image))

        return augmented_images