main.py

import numpy as np
import os
import cv2
from PIL import Image
from matplotlib import pyplot as plt

from config.config import Config
from data.dataLoader import dataLoader

if __name__ == '__main__':
    dl = dataLoader(Config.DATASET_PATH)
    images, captions = dl.get_all()
    print('Number of images:', len(images))
    print('Number of captions:', len(captions))