Spaces:
Running on Zero
Running on Zero
solrz commited on
Commit ·
1ed0595
1
Parent(s): 8eac530
Increase video pointmap mesh density
Browse files
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 =
|
| 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-
|
| 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]))),
|