-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdata_config.py
24 lines (19 loc) · 1.06 KB
/
data_config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
### NF2VEC DATA
NF2VEC_EMB_PATH = "" # NeRF embeddings produced by nf2vec
NF2VEC_DATA_PATH = "" # ShapeNetRender dataset (views + NeRF weights)
NF2VEC_CKPT_PATH = "" # weights of trained nf2vec framework
### OUR DATA, i.e. (nf2vec_emb, clip_emb) pairs
EMB_IMG_PATH = "" # CLIP embeddings of NeRF ground-truth views (one .h5 file for each batch of pairs)
EMB_IMG_SPLIT_PATH = "" # CLIP embeddings of NeRF ground-truth views (one .h5 file for each pair)
EMB_IMG_RENDER_PATH = "" # CLIP embeddings of NeRF rendered views (one .h5 file for each batch of pairs)
EMB_IMG_RENDER_SPLIT_PATH = "" # CLIP embeddings of NeRF rendered views (one .h5 file for each pair)
EMB_TEXT_PATH = "" # CLIP embeddings of BLIP-2 captions
EMB_CONTROLNET_PATH = "" # CLIP embeddings of ControlNet-generated images
BLIP2_CAPTIONS_PATH = "" # BLIP-2 captions of ShapeNetRender
TRAIN_SPLIT = "train"
VAL_SPLIT = "val"
TEST_SPLIT = "test"
DOMAINNET_PATH = ""
SHAPENET_DEPTH_PATH = "" # ShapeNetRender depth-maps
N_VIEWS_MEAN = 36
N_VIEWS_CONTROLNET = 7