UncleanCode commited on
Commit
4ddbb00
·
verified ·
1 Parent(s): be867cc

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
- sys.path.insert(0, '/home/dev/polymorphic/vendor/AnySplat')
 
 
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 = Path('/home/dev/polymorphic')
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)