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"
OCR_TASK_NAME = "OCR"
OCR_WITH_REGION_TASK_NAME = "OCR with Region"

TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME,
    OCR_TASK_NAME,
    OCR_WITH_REGION_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>",
    OCR_TASK_NAME: "<OCR>",
    OCR_WITH_REGION_TASK_NAME: "<OCR_WITH_REGION>"
}
CAPTION_TASK_NAMES = [
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME
]