02alexander commited on
Commit
7861e64
·
1 Parent(s): c9ff1c2

skip generation of video

Browse files
Files changed (1) hide show
  1. app.py +19 -19
app.py CHANGED
@@ -245,26 +245,26 @@ def _make3d(output_queue: SimpleQueue, images: Image.Image):
245
  print(f'type(planes)={type(planes)}')
246
 
247
  # # get video
248
- chunk_size = 20 if IS_FLEXICUBES else 1
249
- render_size = 384
250
 
251
- # frames = []
252
- for i in tqdm(range(0, render_cameras.shape[1], chunk_size)):
253
- if IS_FLEXICUBES:
254
- frame = model.forward_geometry(
255
- planes,
256
- render_cameras[:, i:i+chunk_size],
257
- render_size=render_size,
258
- )['img']
259
- else:
260
- frame = model.synthesizer(
261
- planes,
262
- cameras=render_cameras[:, i:i+chunk_size],
263
- render_size=render_size,
264
- )['images_rgb']
265
-
266
- print(f'type(frame)={type(frame)}')
267
- output_queue.put(("log", "3dvideo", rr.Image(frame)))
268
  # frames.append(frame)
269
 
270
  # frames = torch.cat(frames, dim=1)
 
245
  print(f'type(planes)={type(planes)}')
246
 
247
  # # get video
248
+ # chunk_size = 20 if IS_FLEXICUBES else 1
249
+ # render_size = 384
250
 
251
+ # # frames = []
252
+ # for i in tqdm(range(0, render_cameras.shape[1], chunk_size)):
253
+ # if IS_FLEXICUBES:
254
+ # frame = model.forward_geometry(
255
+ # planes,
256
+ # render_cameras[:, i:i+chunk_size],
257
+ # render_size=render_size,
258
+ # )['img']
259
+ # else:
260
+ # frame = model.synthesizer(
261
+ # planes,
262
+ # cameras=render_cameras[:, i:i+chunk_size],
263
+ # render_size=render_size,
264
+ # )['images_rgb']
265
+
266
+ # print(f'type(frame)={type(frame)}')
267
+ # output_queue.put(("log", "3dvideo", rr.Image(frame)))
268
  # frames.append(frame)
269
 
270
  # frames = torch.cat(frames, dim=1)