gabrielsemiceki9 commited on
Commit
37597a2
·
verified ·
1 Parent(s): d740e61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -2
app.py CHANGED
@@ -1,3 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
  import torch
3
  import torch.nn as nn
@@ -101,6 +115,8 @@ markdown=f'''
101
  - [Project Page](https://scene-dreamer.github.io/)
102
  - [arXiv Link](https://arxiv.org/abs/2302.01330)
103
  Licensed under the S-Lab License.
 
 
104
  We offer a sampled scene whose BEVs are shown on the right. You can also use the button "Generate BEV" to randomly sample a new 3D world represented by a height map and a semantic map. But it requires a long time.
105
 
106
  To render video, push the button "Render" to generate a camera trajectory flying through the world. You can specify rendering options as shown below!
@@ -113,9 +129,9 @@ with gr.Blocks() as demo:
113
  with gr.Column():
114
  with gr.Row():
115
  with gr.Column():
116
- semantic = gr.Image(value='./test/colormap.png',type="pil", shape=(512, 512))
117
  with gr.Column():
118
- height = gr.Image(value='./test/heightmap.png', type="pil", shape=(512, 512))
119
  with gr.Row():
120
  # with gr.Column():
121
  # image = gr.Image(type='pil', shape(540, 960))
 
1
+ import os
2
+ import sys
3
+ import html
4
+ import glob
5
+ import uuid
6
+ import hashlib
7
+ import requests
8
+ from tqdm import tqdm
9
+
10
+ os.system("git clone https://github.com/FrozenBurning/SceneDreamer.git")
11
+ os.system("cp -r SceneDreamer/* ./")
12
+ os.system("bash install.sh")
13
+
14
+
15
  import os
16
  import torch
17
  import torch.nn as nn
 
115
  - [Project Page](https://scene-dreamer.github.io/)
116
  - [arXiv Link](https://arxiv.org/abs/2302.01330)
117
  Licensed under the S-Lab License.
118
+
119
+
120
  We offer a sampled scene whose BEVs are shown on the right. You can also use the button "Generate BEV" to randomly sample a new 3D world represented by a height map and a semantic map. But it requires a long time.
121
 
122
  To render video, push the button "Render" to generate a camera trajectory flying through the world. You can specify rendering options as shown below!
 
129
  with gr.Column():
130
  with gr.Row():
131
  with gr.Column():
132
+ semantic = gr.Image(value='./test/colormap.png',type="pil", height=512, width=512)
133
  with gr.Column():
134
+ height = gr.Image(value='./test/heightmap.png', type="pil", height=512, width=512)
135
  with gr.Row():
136
  # with gr.Column():
137
  # image = gr.Image(type='pil', shape(540, 960))