unilm/kosmos-g/eval/dreambench_prompts.py

OBJECT = {
    "backpack": "backpack",
    "backpack_dog": "backpack",
    "bear_plushie": "stuffed animal",
    "berry_bowl": "bowl",
    "can": "can",
    "candle": "candle",
    "clock": "clock",
    "colorful_sneaker": "sneaker",
    "duck_toy": "toy",
    "fancy_boot": "boot",
    "grey_sloth_plushie": "stuffed animal",
    "monster_toy": "toy",
    "pink_sunglasses": "glasses",
    "poop_emoji": "toy",
    "rc_car": "toy",
    "red_cartoon": "cartoon",
    "robot_toy": "toy",
    "shiny_sneaker": "sneaker",
    "teapot": "teapot",
    "vase": "vase",
    "wolf_plushie": "stuffed animal"
}

LIVE_OBJECT = {
    "cat": "cat",
    "cat2": "cat",
    "dog": "dog",
    "dog2": "dog",
    "dog3": "dog",
    "dog5": "dog",
    "dog6": "dog",
    "dog7": "dog",
    "dog8": "dog",
}

OBJECT_PROMPTS = [
    'a {} in the jungle',
    'a {} in the snow',
    'a {} on the beach',
    'a {} on a cobblestone street',
    'a {} on top of pink fabric',
    'a {} on top of a wooden floor',
    'a {} with a city in the background',
    'a {} with a mountain in the background',
    'a {} with a blue house in the background',
    'a {} on top of a purple rug in a forest',
    'a {} with a wheat field in the background',
    'a {} with a tree and autumn leaves in the background',
    'a {} with the Eiffel Tower in the background',
    'a {} floating on top of water',
    'a {} floating in an ocean of milk',
    'a {} on top of green grass with sunflowers around it',
    'a {} on top of a mirror',
    'a {} on top of the sidewalk in a crowded street',
    'a {} on top of a dirt road',
    'a {} on top of a white rug',
    'a red {}',
    'a purple {}',
    'a shiny {}',
    'a wet {}',
    'a cube shaped {}'
]

LIVE_OBJECT_PROMPTS = [
    'a {} in the jungle',
    'a {} in the snow',
    'a {} on the beach',
    'a {} on a cobblestone street',
    'a {} on top of pink fabric',
    'a {} on top of a wooden floor',
    'a {} with a city in the background',
    'a {} with a mountain in the background',
    'a {} with a blue house in the background',
    'a {} on top of a purple rug in a forest',
    'a {} wearing a red hat',
    'a {} wearing a santa hat',
    'a {} wearing a rainbow scarf',
    'a {} wearing a black top hat and a monocle',
    'a {} in a chef outfit',
    'a {} in a firefighter outfit',
    'a {} in a police outfit',
    'a {} wearing pink glasses',
    'a {} wearing a yellow shirt',
    'a {} in a purple wizard outfit',
    'a red {}',
    'a purple {}',
    'a shiny {}',
    'a wet {}',
    'a cube shaped {}'
]

KOSMOSG_OBJECT_PROMPTS = [
    '{} in the jungle',
    '{} in the snow',
    '{} on the beach',
    '{} on a cobblestone street',
    '{} on top of pink fabric',
    '{} on top of a wooden floor',
    '{} with a city in the background',
    '{} with a mountain in the background',
    '{} with a blue house in the background',
    '{} on top of a purple rug in a forest',
    '{} with a wheat field in the background',
    '{} with a tree and autumn leaves in the background',
    '{} with the Eiffel Tower in the background',
    '{} floating on top of water',
    '{} floating in an ocean of milk',
    '{} on top of green grass with sunflowers around it',
    '{} on top of a mirror',
    '{} on top of the sidewalk in a crowded street',
    '{} on top of a dirt road',
    '{} on top of a white rug',
    '{}, red',
    '{}, purple',
    '{}, shiny',
    '{}, wet',
    '{}, cube shaped'
]

KOSMOSG_LIVE_OBJECT_PROMPTS = [
    '{} in the jungle',
    '{} in the snow',
    '{} on the beach',
    '{} on a cobblestone street',
    '{} on top of pink fabric',
    '{} on top of a wooden floor',
    '{} with a city in the background',
    '{} with a mountain in the background',
    '{} with a blue house in the background',
    '{} on top of a purple rug in a forest',
    '{} wearing a red hat',
    '{} wearing a santa hat',
    '{} wearing a rainbow scarf',
    '{} wearing a black top hat and a monocle',
    '{} in a chef outfit',
    '{} in a firefighter outfit',
    '{} in a police outfit',
    '{} wearing pink glasses',
    '{} wearing a yellow shirt',
    '{} in a purple wizard outfit',
    '{}, red',
    '{}, purple',
    '{}, shiny',
    '{}, wet',
    '{}, cube shaped'
]