from configs.utils import *


# Create input files (along with word map)

create_input_files(dataset = DATASET,
                   json_path= get_captions_path(DATASET),  # path of the .json file with the captions
                   image_folder= get_images_path(DATASET),  # folder containing the images
                   captions_per_image=5,
                   min_word_freq=2,
                   output_folder=get_path(ARCHITECTURE, input=True, fine_tune=fine_tune_encoder),
                   max_len=30)

if __name__ == "__main__":
    create_input_files()