Image-to-3D
3d
dmitriitochilkin commited on
Commit
c1cf771
1 Parent(s): 41e168b

update model

Browse files
Files changed (2) hide show
  1. config.yaml +38 -0
  2. tripo_sr.ckpt → model.ckpt +2 -2
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
tripo_sr.ckpt → model.ckpt RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9d494605b1fffe67e9d09f53a803714bc5dacbd061fcdcc2311f95532a05dcd7
3
- size 1904195926
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:429e2c6b22a0923967459de24d67f05962b235f79cde6b032aa7ed2ffcd970ee
3
+ size 1677246742