File size: 1,890 Bytes
ad1357a 6f01b45 ad1357a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
# config.py
# 配置相关：API、场景等
import os
BACKEND_URL = os.getenv("BACKEND_URL", "http://47.95.6.204:51001")
API_ENDPOINTS = {
"submit_task": f"{BACKEND_URL}/predict/video",
"query_status": f"{BACKEND_URL}/predict/task",
"get_result": f"{BACKEND_URL}//predict"
}
SCENE_CONFIGS = {
"demo1": {
"description": "Demo 1",
"objects": ["bedroom", "kitchen", "living room", ""],
"preview_image": "./assets/scene_1.png",
"default_instruction": "Walk past the left side of the bed and stop in the doorway."
},
"demo2": {
"description": "Demo 2",
"objects": ["office", "meeting room", "corridor"],
"preview_image": "./assets/scene_2.png",
"default_instruction": "Walk through the bathroom, past the sink and toilet. Stop in front of the counter with the two suitcase."
},
"demo3": {
"description": "Demo 3",
"objects": ["garage", "workshop", "storage"],
"preview_image": "./assets/scene_3.png",
"default_instruction": "Do a U-turn. Walk forward through the kitchen, heading to the black door. Walk out of the door and take a right onto the deck. Walk out on to the deck and stop."
},
"demo4": {
"description": "Demo 4",
"objects": ["garden", "patio", "pool"],
"preview_image": "./assets/scene_4.png",
"default_instruction": "Walk out of bathroom and stand on white bath mat."
},
"demo5": {
"description": "Demo 5",
"objects": ["library", "hall", "lounge"],
"preview_image": "./assets/scene_5.png",
"default_instruction": "Walk straight through the double wood doors, follow the red carpet straight to the next doorway and stop where the carpet splits off."
},
}
MODEL_CHOICES = ["rdp", "cma"]
MODE_CHOICES = ["vlnPE", "vlnCE"]
|