Image-to-3D
3d
hwjiang commited on
Commit
c91a29c
1 Parent(s): eee9ff0

Upload config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +38 -0
config.yaml ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ cond_image_size: 512
2
+
3
+ image_tokenizer_cls: tsr.models.tokenizers.image.DINOSingleImageTokenizer
4
+ image_tokenizer:
5
+ pretrained_model_name_or_path: "facebook/dino-vitb16"
6
+
7
+ tokenizer_cls: tsr.models.tokenizers.triplane.Triplane1DTokenizer
8
+ tokenizer:
9
+ plane_size: 32
10
+ num_channels: 1024
11
+
12
+ backbone_cls: tsr.models.transformer.transformer_1d.Transformer1D
13
+ backbone:
14
+ in_channels: ${tokenizer.num_channels}
15
+ num_attention_heads: 16
16
+ attention_head_dim: 64
17
+ num_layers: 16
18
+ cross_attention_dim: 768
19
+
20
+ post_processor_cls: tsr.models.network_utils.TriplaneUpsampleNetwork
21
+ post_processor:
22
+ in_channels: 1024
23
+ out_channels: 40
24
+
25
+ decoder_cls: tsr.models.network_utils.NeRFMLP
26
+ decoder:
27
+ in_channels: 120 # 3 * 40
28
+ n_neurons: 64
29
+ n_hidden_layers: 9
30
+ activation: silu
31
+
32
+ renderer_cls: tsr.models.nerf_renderer.TriplaneNeRFRenderer
33
+ renderer:
34
+ radius: 0.87 # slightly larger than 0.5 * sqrt(3)
35
+ feature_reduction: concat
36
+ density_activation: exp
37
+ density_bias: -1.0
38
+ num_samples_per_ray: 128