utils/tasks.py

OBJECT_DETECTION_TASK_NAME = "Object Detection"
CAPTION_TASK_NAME = "Caption"
DETAILED_CAPTION_TASK_NAME = "Detailed Caption"
MORE_DETAILED_CAPTION_TASK_NAME = "More Detailed Caption"

TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME
]
TASKS = {
    OBJECT_DETECTION_TASK_NAME: "<OD>",
    CAPTION_TASK_NAME: "<CAPTION>",
    DETAILED_CAPTION_TASK_NAME: "<DETAILED_CAPTION>",
    MORE_DETAILED_CAPTION_TASK_NAME: "<MORE_DETAILED_CAPTION>"
}
CAPTION_TASK_NAMES = [
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME
]