draw rectangle

from PIL import Image, ImageDraw
import cv2

def draw_rect(img, box, img_src):
    x1, y1, x2, y2 = map(int, box)
    if img_src == 'pil':
        draw = ImageDraw.Draw(img)
        draw.rectangle(((x1, y1), (x2, y2)), width=10, outline="red")
        del draw
    elif img_src == 'cv2':
        img = cv2.rectangle(img, (x1, y1), (x2, y2), (0, 0, 255), 4)        
    return img