update

app.py

@@ -46,6 +46,11 @@ RT = torch.tensor([[ -1.0000,   0.0000,   0.0000,   0.0000,   0.0000,   0.5000,
 
 obj_dict = {}
 
+# init
+fake_bevs = torch.zeros([1, 14, 256, 256], device='cuda').float()
+_ = G(code, RT, fake_bevs)
+
+
 def trans(x, y, z, length):
     w = h = length
     x = 0.5 * w - 128 + 256 - (x/9 + .5) * 256
@@ -105,6 +110,7 @@ def objs_to_canvas(lst, length=256, scale = 2.6):
 def predict_local_view(lst):
     canvas = torch.tensor(objs_to_canvas(lst)).cuda()[None]
     bevs = canvas[..., 0: 0+256]
+    print(code.shape, RT.shape, bevs.shape)
     gen = G(code, RT, bevs)
     rgb = gen['gen_output']['image'][0] * .5 + .5
     return to_pil(rgb)