Upload via Autoresearch export_data
Browse files
experiments/fourier-splatting/scripts/make_fourier_scene_generic.py
CHANGED
|
@@ -4,13 +4,15 @@ import numpy as np
|
|
| 4 |
import torch
|
| 5 |
from PIL import Image
|
| 6 |
from plyfile import PlyData, PlyElement
|
| 7 |
-
|
|
|
|
|
|
|
| 8 |
from src.model.model.anysplat import AnySplat
|
| 9 |
from src.utils.image import process_image
|
| 10 |
|
| 11 |
if len(sys.argv) != 3:
|
| 12 |
raise SystemExit('usage: make_fourier_scene_generic.py FRAMES_DIR OUTPUT_DIR')
|
| 13 |
-
ROOT =
|
| 14 |
FRAMES = sorted(Path(sys.argv[1]).glob('*.jpg'))
|
| 15 |
OUT = Path(sys.argv[2])
|
| 16 |
OUT.mkdir(parents=True, exist_ok=True)
|
|
|
|
| 4 |
import torch
|
| 5 |
from PIL import Image
|
| 6 |
from plyfile import PlyData, PlyElement
|
| 7 |
+
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
| 8 |
+
ANYSPLAT_ROOT = PROJECT_ROOT / 'src' / 'AnySplat'
|
| 9 |
+
sys.path.insert(0, str(ANYSPLAT_ROOT))
|
| 10 |
from src.model.model.anysplat import AnySplat
|
| 11 |
from src.utils.image import process_image
|
| 12 |
|
| 13 |
if len(sys.argv) != 3:
|
| 14 |
raise SystemExit('usage: make_fourier_scene_generic.py FRAMES_DIR OUTPUT_DIR')
|
| 15 |
+
ROOT = PROJECT_ROOT
|
| 16 |
FRAMES = sorted(Path(sys.argv[1]).glob('*.jpg'))
|
| 17 |
OUT = Path(sys.argv[2])
|
| 18 |
OUT.mkdir(parents=True, exist_ok=True)
|