solrz commited on
Commit
1ed0595
·
1 Parent(s): 8eac530

Increase video pointmap mesh density

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,7 +70,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
70
  MAX_HEIGHT = 1024 # cap input height before processing — keeps everything fast
71
  VIDEO_MAX_HEIGHT = 512
72
  VIDEO_DEFAULT_FRAMES = 36
73
- VIDEO_DEFAULT_STRIDE = 3
74
 
75
  _fg_transform = transforms.Compose([
76
  transforms.Resize((1024, 768)),
@@ -370,7 +370,7 @@ def _point_sequence_zip(
370
  positions = np.stack(positions_frames, axis=0).astype(np.float32)
371
  colors = np.stack(colors_frames, axis=0).astype(np.uint8)
372
  metadata = {
373
- "format": "fpbox-sapiens-pointmap-sequence-v3",
374
  "model": f"sapiens2-pointmap-{size}",
375
  "frameCount": int(positions.shape[0]),
376
  "fps": float(min(fps, max(1, positions.shape[0]))),
 
70
  MAX_HEIGHT = 1024 # cap input height before processing — keeps everything fast
71
  VIDEO_MAX_HEIGHT = 512
72
  VIDEO_DEFAULT_FRAMES = 36
73
+ VIDEO_DEFAULT_STRIDE = 2
74
 
75
  _fg_transform = transforms.Compose([
76
  transforms.Resize((1024, 768)),
 
370
  positions = np.stack(positions_frames, axis=0).astype(np.float32)
371
  colors = np.stack(colors_frames, axis=0).astype(np.uint8)
372
  metadata = {
373
+ "format": "fpbox-sapiens-pointmap-sequence-v4",
374
  "model": f"sapiens2-pointmap-{size}",
375
  "frameCount": int(positions.shape[0]),
376
  "fps": float(min(fps, max(1, positions.shape[0]))),