utils/tasks.py

OBJECT_DETECTION_TASK_NAME = "Object Detection"
REGION_PROPOSAL_TASK_NAME = "Region Proposal"
DENSE_REGION_CAPTION_TASK_NAME = "Dense Region Caption"
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"
REGION_TO_CATEGORY_TASK_NAME = "Region to Category"
REGION_TO_DESCRIPTION_TASK_NAME = "Region to Description"

TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    REGION_PROPOSAL_TASK_NAME,
    DENSE_REGION_CAPTION_TASK_NAME,
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME,
    OCR_TASK_NAME,
    OCR_WITH_REGION_TASK_NAME,
    REGION_TO_CATEGORY_TASK_NAME,
    REGION_TO_DESCRIPTION_TASK_NAME
]
TASKS = {
    OBJECT_DETECTION_TASK_NAME: "<OD>",
    REGION_PROPOSAL_TASK_NAME: "<REGION_PROPOSAL>",
    DENSE_REGION_CAPTION_TASK_NAME: "<DENSE_REGION_CAPTION>",
    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>",
    REGION_TO_CATEGORY_TASK_NAME: "<REGION_TO_CATEGORY>",
    REGION_TO_DESCRIPTION_TASK_NAME: "<REGION_TO_DESCRIPTION>"
}
IMAGE_INPUT_TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    REGION_PROPOSAL_TASK_NAME,
    DENSE_REGION_CAPTION_TASK_NAME,
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME,
    OCR_TASK_NAME,
    OCR_WITH_REGION_TASK_NAME,
]
IMAGE_PROMPTER_INPUT_TASK_NAMES = [
    REGION_TO_CATEGORY_TASK_NAME,
    REGION_TO_DESCRIPTION_TASK_NAME
]
IMAGE_OUTPUT_TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    REGION_PROPOSAL_TASK_NAME,
    DENSE_REGION_CAPTION_TASK_NAME,
    OCR_WITH_REGION_TASK_NAME,
    REGION_TO_CATEGORY_TASK_NAME,
    REGION_TO_DESCRIPTION_TASK_NAME
]
TEXTBOX_OUTPUT_TASK_NAMES = [
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME,
    OCR_TASK_NAME
]
IMAGE_TO_IMAGE_TASK_NAMES = [
    OBJECT_DETECTION_TASK_NAME,
    OCR_WITH_REGION_TASK_NAME,
    REGION_PROPOSAL_TASK_NAME,
    DENSE_REGION_CAPTION_TASK_NAME
]
IMAGE_TO_TEXT_TASK_NAMES = [
    CAPTION_TASK_NAME,
    DETAILED_CAPTION_TASK_NAME,
    MORE_DETAILED_CAPTION_TASK_NAME,
    OCR_TASK_NAME
]
IMAGE_PROMPT_TO_IMAGE_TASK_NAMES = [
    REGION_TO_CATEGORY_TASK_NAME,
    REGION_TO_DESCRIPTION_TASK_NAME
]