Create constants.py
Browse files- constants.py +32 -0
constants.py
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
CONTROLLER_HEART_BEAT_EXPIRATION = 30
|
| 2 |
+
WORKER_HEART_BEAT_INTERVAL = 15
|
| 3 |
+
|
| 4 |
+
LOGDIR = "."
|
| 5 |
+
|
| 6 |
+
# Model Constants
|
| 7 |
+
IGNORE_INDEX = -100
|
| 8 |
+
IMAGE_TOKEN_INDEX = -200
|
| 9 |
+
DEFAULT_IMAGE_TOKEN = "<image>"
|
| 10 |
+
DEFAULT_IMAGE_PATCH_TOKEN = "<im_patch>"
|
| 11 |
+
DEFAULT_IM_START_TOKEN = "<im_start>"
|
| 12 |
+
DEFAULT_IM_END_TOKEN = "<im_end>"
|
| 13 |
+
IMAGE_PLACEHOLDER = "<image-placeholder>"
|
| 14 |
+
|
| 15 |
+
# Added by Ferret
|
| 16 |
+
DEFAULT_REGION_FEA_TOKEN = "<region_fea>"
|
| 17 |
+
VOCAB_IMAGE_W = 1000
|
| 18 |
+
VOCAB_IMAGE_H = 1000
|
| 19 |
+
|
| 20 |
+
# GROUNDING PROMPTS
|
| 21 |
+
GROUNDING_TEMPLATES = [
|
| 22 |
+
'\nProvide the bounding boxes of the mentioned objects.',
|
| 23 |
+
'\nInclude the coordinates for each mentioned object.',
|
| 24 |
+
'\nLocate the objects with their coordinates.',
|
| 25 |
+
'\nAnswer in [x1, y1, x2, y2] format.',
|
| 26 |
+
'\nMention the objects and their locations using the format [x1, y1, x2, y2].',
|
| 27 |
+
'\nDraw boxes around the mentioned objects.',
|
| 28 |
+
'\nUse boxes to show where each thing is.',
|
| 29 |
+
'\nTell me where the objects are with coordinates.',
|
| 30 |
+
'\nList where each object is with boxes.',
|
| 31 |
+
'\nShow me the regions with boxes.'
|
| 32 |
+
]
|