-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpixi.toml
76 lines (67 loc) · 2.47 KB
/
pixi.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
[project]
name = "simplerecon"
version = "0.1.0"
description = "Add a short description here"
authors = ["Pablo Vela <[email protected]>"]
channels = ["nvidia", "pytorch", "conda-forge"]
platforms = ["linux-64", "osx-arm64"]
[system-requirements]
macos = "11.0" # needed for some reason otherwise fails to resolve mediapipe package
libc = "2.31"
[tasks]
download-hero-model = """
bash -c '[ ! -f \"weights/hero_model.ckpt\" ] && gdown 1hCuKZjEq-AghrYAmFxJs_4eeixIlP488 -O weights/'
"""
download-vdr-data = """
bash -c '[ ! -d \"data/vdr\" ] && (gdown 1x-auV7vGCMdu5yZUMPcoP83p77QOuasT -O data/ && unzip data/vdr.zip -d data/) || echo \"Dataset already downloaded.\"'
"""
download-data = {cmd="pwd", depends_on = ["download-hero-model", "download-vdr-data"]}
[dependencies]
python = "3.9.*"
pytorch = "1.13.1.*"
pytorch-lightning = "1.5.4.*"
scipy = ">=1.11.4,<1.12"
torchvision = "0.14.1.*"
[pypi-dependencies]
Pillow = ">=9.4.0,<9.5"
antialiased-cnns = "*"
efficientnet_pytorch = "*"
einops = "*" # batching one liners
gdown = ">=5.2.0"
kornia = "==0.6.7" # gradients
matplotlib = ">=3.8.2,<3.9"
moviepy = "*"
open3d = "*" # mesh fusion
pip = "*"
pyrender = "*"# rendering meshes
scikit-image = "*"
tensorboard = ">=2.15.1,<2.16"
timm = "*" # efficent
transforms3d = "*" # for NeuralRecon's arkit
trimesh = "*" # mesh loading/storage, and mesh generation
rerun-sdk = "==0.16.0"
[target.linux-64.dependencies]
pytorch-cuda = "11.7.*"
[target.osx-arm64.tasks]
# OMP_NUM_THREADS required to work on mac https://github.com/isl-org/Open3D/issues/6196
vdr = """OMP_NUM_THREADS=1 python rerun_visualize_live_meshing.py \
--name HERO_MODEL \
--output_base_path OUTPUT_PATH \
--config_file configs/models/hero_model.yaml \
--load_weights_from_checkpoint weights/hero_model.ckpt \
--data_config_file configs/data/vdr_dense_rerun.yaml \
--num_workers 8 \
--run_fusion \
--depth_fuser open3d \
--fuse_color"""
[target.linux-64.tasks]
vdr = """CUDA_VISIBLE_DEVICES=0 python rerun_visualize_live_meshing.py \
--name HERO_MODEL \
--output_base_path OUTPUT_PATH \
--config_file configs/models/hero_model.yaml \
--load_weights_from_checkpoint weights/hero_model.ckpt \
--data_config_file configs/data/vdr_dense_rerun.yaml \
--num_workers 8 \
--run_fusion \
--depth_fuser open3d \
--fuse_color"""