yuqiang6149 commited on
Commit
4973bd2
·
verified ·
1 Parent(s): 7195a77

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +72 -72
config.py CHANGED
@@ -1,72 +1,72 @@
1
- # config.py
2
- # 配置相关:API、场景等
3
- import os
4
-
5
- BACKEND_URL = os.getenv("BACKEND_URL", "http://47.95.6.204:51001")
6
- API_ENDPOINTS = {
7
- "submit_task": f"{BACKEND_URL}/predict/video",
8
- "query_status": f"{BACKEND_URL}/predict/task",
9
- "get_result": f"{BACKEND_URL}//predict"
10
- }
11
-
12
- SCENE_CONFIGS = {
13
- "demo1": {
14
- "description": "Demo 1",
15
- "objects": ["bedroom", "kitchen", "living room", ""],
16
- "preview_image": "./assets/scene_1.png",
17
- #"glb_path": "scene_assets/scene1_no_ceiling.glb",
18
- "glb_path": "scene_assets/simply_3d.obj",
19
- "scene_name": "17DRP5sb8fy",
20
- "default_instruction": "Walk past the left side of the bed and stop in the doorway."
21
- },
22
- "demo2": {
23
- "description": "Demo 2",
24
- "objects": ["office", "meeting room", "corridor"],
25
- "preview_image": "./assets/scene_2.png",
26
- "glb_path": "scene_assets/scene2_no_ceiling.glb",
27
- "scene_name": "r1Q1Z4BcV1o",
28
- "default_instruction": "Walk through the bathroom, past the sink and toilet. Stop in front of the counter with the two suitcase."
29
- },
30
- "demo3": {
31
- "description": "Demo 3",
32
- "objects": ["garage", "workshop", "storage"],
33
- "preview_image": "./assets/scene_3.png",
34
- "glb_path": "scene_assets/scene3_no_ceiling.glb",
35
- "scene_name": "dhjEzFoUFzH",
36
- "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."
37
- },
38
- "demo4": {
39
- "description": "Demo 4",
40
- "objects": ["garden", "patio", "pool"],
41
- "preview_image": "./assets/scene_4.png",
42
- "glb_path": "scene_assets/scene4_no_ceiling.glb",
43
- "scene_name": "demo4", # 没有对应的图片文件,保持原名
44
- "default_instruction": "Walk out of bathroom and stand on white bath mat."
45
- },
46
- "demo5": {
47
- "description": "Demo 5",
48
- "objects": ["library", "hall", "lounge"],
49
- "preview_image": "./assets/scene_5.png",
50
- "glb_path": "scene_assets/scene4_no_ceiling.glb", # 只有scene1-4的GLB文件
51
- "scene_name": "demo5", # 没有对应的图片文件,保持原名
52
- "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."
53
- },
54
- }
55
-
56
- EPISODE_CONFIGS = {
57
- "episode_1": {
58
- "description": "1",
59
- },
60
- "episode_2": {
61
- "description": "2",
62
- },
63
- "episode_3": {
64
- "description": "3",
65
- },
66
- "episode_4": {
67
- "description": "4",
68
- }
69
- }
70
-
71
- MODEL_CHOICES = ["rdp", "cma"]
72
- MODE_CHOICES = ["vlnPE", "vlnCE"]
 
1
+ # config.py
2
+ # 配置相关:API、场景等
3
+ import os
4
+
5
+ BACKEND_URL = os.getenv("BACKEND_URL", "http://47.95.6.204:51001")
6
+ API_ENDPOINTS = {
7
+ "submit_task": f"{BACKEND_URL}/predict/video",
8
+ "query_status": f"{BACKEND_URL}/predict/task",
9
+ "get_result": f"{BACKEND_URL}//predict"
10
+ }
11
+
12
+ SCENE_CONFIGS = {
13
+ "demo1": {
14
+ "description": "Demo 1",
15
+ "objects": ["bedroom", "kitchen", "living room", ""],
16
+ "preview_image": "./assets/scene_1.png",
17
+ "glb_path": "scene_assets/scene1_no_ceiling_compressed.glb",
18
+ #"glb_path": "scene_assets/simply_3d.obj",
19
+ "scene_name": "17DRP5sb8fy",
20
+ "default_instruction": "Walk past the left side of the bed and stop in the doorway."
21
+ },
22
+ "demo2": {
23
+ "description": "Demo 2",
24
+ "objects": ["office", "meeting room", "corridor"],
25
+ "preview_image": "./assets/scene_2.png",
26
+ "glb_path": "scene_assets/scene2_no_ceiling_compressed.glb",
27
+ "scene_name": "r1Q1Z4BcV1o",
28
+ "default_instruction": "Walk through the bathroom, past the sink and toilet. Stop in front of the counter with the two suitcase."
29
+ },
30
+ "demo3": {
31
+ "description": "Demo 3",
32
+ "objects": ["garage", "workshop", "storage"],
33
+ "preview_image": "./assets/scene_3.png",
34
+ "glb_path": "scene_assets/scene3_no_ceiling_compressed.glb",
35
+ "scene_name": "dhjEzFoUFzH",
36
+ "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."
37
+ },
38
+ "demo4": {
39
+ "description": "Demo 4",
40
+ "objects": ["garden", "patio", "pool"],
41
+ "preview_image": "./assets/scene_4.png",
42
+ "glb_path": "scene_assets/scene4_no_ceiling_compressed.glb",
43
+ "scene_name": "demo4", # 没有对应的图片文件,保持原名
44
+ "default_instruction": "Walk out of bathroom and stand on white bath mat."
45
+ },
46
+ "demo5": {
47
+ "description": "Demo 5",
48
+ "objects": ["library", "hall", "lounge"],
49
+ "preview_image": "./assets/scene_5.png",
50
+ "glb_path": "scene_assets/scene4_no_ceiling_compressed.glb", # 只有scene1-4的GLB文件
51
+ "scene_name": "demo5", # 没有对应的图片文件,保持原名
52
+ "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."
53
+ },
54
+ }
55
+
56
+ EPISODE_CONFIGS = {
57
+ "episode_1": {
58
+ "description": "1",
59
+ },
60
+ "episode_2": {
61
+ "description": "2",
62
+ },
63
+ "episode_3": {
64
+ "description": "3",
65
+ },
66
+ "episode_4": {
67
+ "description": "4",
68
+ }
69
+ }
70
+
71
+ MODEL_CHOICES = ["rdp", "cma"]
72
+ MODE_CHOICES = ["vlnPE", "vlnCE"]