import albumentations as a
import numpy as np
from os import listdir as ls
import cv2 as cv
from sys import argv, exit

if len(argv) != 2:
    exit(1)

def b_c(b, s = (1920, 1080)):
    return [
        round(b[0] * s[0] - b[2] * s[0] / 2),
        round(b[0] * s[0] + b[2] * s[0] / 2),
        round(b[1] * s[1] - b[3] * s[1] / 2),
        round(b[1] * s[1] + b[3] * s[1] / 2),
    ]

def sq(img, bbox):
    for i in bbox:
        b = b_c(i, s = (img.shape[1], img.shape[0]))
        #print(b)
        cv.rectangle(img, (b[0], b[2]), (b[1], b[3]), (255, 0, 0), 2)

img = cv.cvtColor(cv.imread(f"{argv[1]}.jpg"), cv.COLOR_BGR2RGB)
box = [tuple(map(float, j.split()[1:])) for j in open(f"{argv[1]}.txt").read().split("\n") if j]

box = np.array(box)

sq(img, box)

from matplotlib import pyplot as plt
plt.imshow(img)
plt.show()