hackathonf23-Creativity_Underflow/settings.py

class TrainSettings:
    ## Paths
    class Dataset:
        # path to input images
        path_to_images = "./data/train2017/train2017"

        # path to pkl file generate by text_preprocessing_sng_parser.py
        rule_parser_path = "./train_data/captions_preprocessed.pkl"

        # path to coreNLP json file
        coreNLP_path = "./train_data/captions_preprocessed.json"

        # path to annotations (captions)
        annotations_path = "./data/annotations_trainval2017/annotations/captions_train2017.json"

    class Output:
        out_dir = "./train_data"

    class Logging:
        project_name = "WhatIsGoingOn"
        model_dir = "./models"

class ValidationSettings:
    ## Paths
    class Dataset:
        # path to input images
        path_to_images = "./data/val2017/val2017"

        # path to pkl file generate by text_preprocessing_sng_parser.py
        rule_parser_path = "./val_data/captions_preprocessed.pkl"

        # path to coreNLP json file
        coreNLP_path = "./val_data/captions_preprocessed.json"

        # path to annotations (captions)
        annotations_path = "./data/annotations_trainval2017/annotations/captions_val2017.json"

    class Output:
        out_dir = "./val_data"