from pathlib import Path
import cv2
import os.path

from dto.annotationClass import AnnotationClass
from dto.imageLabel import ImageLabel
from preprocessing import read_labels

images_dir = '../images'
labels_dir = '../labels'
annotation_classes = AnnotationClass.read_json()


image = os.listdir(images_dir)[0]
image_path = os.path.join(images_dir, image)
labels_path = os.path.join(labels_dir, f'{Path(image_path).stem}.txt')

img = ImageLabel(
    image_path=image_path,
    image=cv2.imread(image_path),
    labels_path=labels_path,
    labels=read_labels(labels_path)
)

img.visualize(annotation_classes)