hongchi commited on
Commit
2e9191e
·
verified ·
1 Parent(s): ebd40a8

Add Fire3D model card

Browse files
Files changed (1) hide show
  1. README.md +105 -0
README.md ADDED
@@ -0,0 +1,105 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ library_name: pytorch
4
+ pipeline_tag: image-to-3d
5
+ tags:
6
+ - 3d-scene-reconstruction
7
+ - rgb-d
8
+ - single-image
9
+ - flow-matching
10
+ - textured-mesh
11
+ ---
12
+
13
+ # FIRE3D: Feed-forward Interactive 3D Scene Reconstruction Within A Minute
14
+
15
+ Fire3D is a unified feed-forward framework that transforms a single RGB image
16
+ or casual RGB video into simulation-ready 3D scene assets. It predicts a
17
+ compositional scene representation with object-level 6-DoF pose, bounding box,
18
+ mesh geometry, and texture, without test-time optimization.
19
+
20
+ This release provides the inference code, model checkpoints, processed example
21
+ inputs, and frozen protocols needed to reproduce Fire3D results on iTHOR,
22
+ Imaginarium, ScanNet++, and single-image scenes.
23
+
24
+ The inference code and installation instructions are maintained at
25
+ [xiahongchi/Fire3D](https://github.com/xiahongchi/Fire3D). Processed release
26
+ inputs are hosted at
27
+ [datasets/hongchi/Fire3D](https://huggingface.co/datasets/hongchi/Fire3D).
28
+
29
+ ## Model Bundle
30
+
31
+ The repository contains the scene perception checkpoint, the three
32
+ flow-matching checkpoints, sparse VAE decoders, latent normalization
33
+ statistics, and the TRELLIS.2 shape/PBR decoders used by the frozen release
34
+ protocols. `manifest.json` records the byte size and SHA-256 digest of every
35
+ inference file; `checksums.sha256` provides the same values in standard form.
36
+
37
+ Fire3D's Hierarchical Compression VAE (HC-VAE) compresses each sparse SC-VAE
38
+ shape and material field from a `32^3` representation with 32 channels into an
39
+ `8^3` representation with 64 channels. This 32x reduction enables batched flow
40
+ sampling and VAE decoding across many scene instances while retaining the
41
+ frozen TRELLIS.2 SC-VAE decoding path.
42
+
43
+ | Component | Bundle path |
44
+ |---|---|
45
+ | Scene perception | `perception/model.pt` |
46
+ | Sparse-structure flow | `reconstruction/flows/ss/model.pt` |
47
+ | Shape flow | `reconstruction/flows/shape/model.pt` |
48
+ | PBR flow | `reconstruction/flows/pbr/model.pt` |
49
+ | Sparse-structure VAE | `reconstruction/vae/ss/ckpts/decoder.pt` |
50
+ | Shape HC-VAE | `reconstruction/vae/shape/ckpts/decoder.pt` |
51
+ | PBR HC-VAE | `reconstruction/vae/pbr/ckpts/decoder.pt` |
52
+ | DINOv3 encoder | `external/dinov3_vitl16_pretrain_lvd1689m-8aa4cbdd.pth` |
53
+ | TRELLIS.2 shape decoder | `external/trellis2/shape_dec_next_dc_f16c32_fp16.safetensors` |
54
+ | TRELLIS.2 PBR decoder | `external/trellis2/tex_dec_next_dc_f16c32_fp16.safetensors` |
55
+
56
+ Public filenames intentionally do not encode private training iteration
57
+ numbers. The current method and HC-VAE figures are available in the
58
+ [GitHub release](https://github.com/xiahongchi/Fire3D#method).
59
+
60
+ The default protocols use classifier-free guidance 3, flow and VAE batches of
61
+ 16, and the batch CuMesh postprocessor. The iTHOR, Imaginarium, and ScanNet++
62
+ protocols enable the validated background room-box fit before background
63
+ reconstruction.
64
+
65
+ ## Use
66
+
67
+ ```bash
68
+ git clone https://github.com/xiahongchi/Fire3D.git
69
+ cd Fire3D
70
+ bash scripts/install.sh
71
+ conda activate fire3d
72
+ bash scripts/install_blender.sh
73
+
74
+ fire3d download --models
75
+ fire3d download --data --dataset ithor --scene-id iTHOR_FloorPlan312_physics
76
+ bash scripts/run_ithor_example.sh
77
+ ```
78
+
79
+ The downloader places this repository under `checkpoints/Fire3D/` and verifies
80
+ the release manifest before inference.
81
+
82
+ ## Scope
83
+
84
+ The released checkpoints are intended for inference through the frozen Fire3D
85
+ protocols. Performance can degrade for scenes with inaccurate camera poses or
86
+ depth, severe occlusion, unusual scale, non-room backgrounds, or objects far
87
+ outside the training distribution. The generated meshes should be reviewed
88
+ before use in safety-critical or physically deployed applications.
89
+
90
+ ## Licenses
91
+
92
+ Original Fire3D code is MIT licensed. This model bundle includes DINOv3 and
93
+ selected TRELLIS.2 files, which retain their upstream licenses. Review the
94
+ included `licenses/` directory and the repository's
95
+ `THIRD_PARTY_NOTICES.md` before use or redistribution.
96
+
97
+ ## Citation
98
+
99
+ ```bibtex
100
+ @misc{xia2026fire3d,
101
+ title={FIRE3D: Feed-forward Interactive 3D Scene Reconstruction Within A Minute},
102
+ author={Hongchi Xia and Tianhang Cheng and Wei-Chiu Ma and Shenlong Wang},
103
+ year={2026}
104
+ }
105
+ ```