estellea commited on
Commit
361d25f
1 Parent(s): ed7758c

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +91 -0
README.md ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: creativeml-openrail-m
3
+ datasets:
4
+ - laion/laion400m
5
+ tags:
6
+ - stable-diffusion
7
+ - stable-diffusion-diffusers
8
+ - text-to-image
9
+ language:
10
+ - en
11
+ ---
12
+
13
+ # LDM3D model
14
+
15
+ The LDM3D model was proposed in ["LDM3D: Latent Diffusion Model for 3D"](https://arxiv.org/abs/2305.10853) by Gabriela Ben Melech Stan, Diana Wofk, Scottie Fox, Alex Redden, Will Saxton, Jean Yu, Estelle Aflalo, Shao-Yen Tseng, Fabio Nonato, Matthias Muller, Vasudev Lal.
16
+
17
+ LDM3D got accepted to [CVPRW'23]([https://aaai.org/Conferences/AAAI-23/](https://cvpr2023.thecvf.com/)).
18
+
19
+ This checkpoint finetunes the previous [ldm3d-4c](https://huggingface.co/Intel/ldm3d-4c) on [polyhaven](https://polyhaven.com/) and [skies]()
20
+ A demo using this checkpoint has been open sourced in [this space](https://huggingface.co/spaces/Intel/ldm3d)
21
+
22
+ ## Model description
23
+
24
+ The abstract from the paper is the following:
25
+ This research paper proposes a Latent Diffusion Model for 3D (LDM3D) that generates both image and depth map data from a given text prompt, allowing users to generate RGBD images from text prompts. The LDM3D model is fine-tuned on a dataset of tuples containing an RGB image, depth map and caption, and validated through extensive experiments. We also develop an application called DepthFusion, which uses the generated RGB images and depth maps to create immersive and interactive 360-degree-view experiences using TouchDesigner. This technology has the potential to transform a wide range of industries, from entertainment and gaming to architecture and design. Overall, this paper presents a significant contribution to the field of generative AI and computer vision, and showcases the potential of LDM3D and DepthFusion to revolutionize content creation and digital experiences.
26
+
27
+ ![LDM3D overview](model_overview.png)
28
+ <font size="2">LDM3D overview taken from [the original paper](https://arxiv.org/abs/2305.10853)</font>
29
+
30
+ ## Intended uses
31
+
32
+ You can use this model to generate RGB and depth map given a text prompt.
33
+ A short video summarizing the approach can be found at [this url](https://t.ly/tdi2) and a VR demo can be found [here](https://www.youtube.com/watch?v=3hbUo-hwAs0).
34
+
35
+
36
+ ### How to use
37
+
38
+ Here is how to use this model to get the features of a given text in PyTorch:
39
+ ```python
40
+
41
+ from diffusers import StableDiffusionLDM3DPipeline
42
+
43
+ pipe = StableDiffusionLDM3DPipeline.from_pretrained("Intel/ldm3d-4c")
44
+ pipe.to("cuda")
45
+
46
+
47
+ prompt ="A picture of some lemons on a table"
48
+ name = "lemons"
49
+
50
+ output = pipe(prompt)
51
+ rgb_image, depth_image = output.rgb, output.depth
52
+ rgb_image[0].save(name+"_ldm3d_rgb.jpg")
53
+ depth_image[0].save(name+"_ldm3d_depth.png")
54
+ ```
55
+
56
+ This is the result:
57
+
58
+ ![ldm3d_results](ldm3d_4c_results.png)
59
+
60
+
61
+ ### Limitations and bias
62
+
63
+ For the image generation, limitations and bias are the same as the ones from [Stable diffusion](https://huggingface.co/CompVis/stable-diffusion-v1-4#limitations)
64
+ For the depth map generation, a first limitiation is that we are using DPT-large to produce the ground truth, hence, other limitations and bias are the same as the ones from [DPT](https://huggingface.co/Intel/dpt-large).
65
+
66
+
67
+ ## Training data
68
+
69
+ The LDM3D model was finetuned on a dataset constructed from a subset of the LAION-400M dataset, a large-scale image-caption dataset that contains over 400 million image-caption pairs.
70
+
71
+ ### Finetuning
72
+
73
+ The fine-tuning process comprises two stages. In the first stage, we train an autoencoder to generate a lower-dimensional, perceptually equivalent data representation. Subsequently, we fine-tune the diffusion model using the frozen autoencoder
74
+
75
+ ## Evaluation results
76
+
77
+ Please refer to Table 1 and Table2 from the [paper](https://arxiv.org/abs/2305.10853) for quantitative results.
78
+ The figure below shows some qualitative results comparing our method with (Stable diffusion v1.4)[https://arxiv.org/pdf/2112.10752.pdf] and with (DPT-Large)[https://arxiv.org/pdf/2103.13413.pdf] for the depth maps
79
+ ![qualitative results](qualitative_results.png)
80
+
81
+ ### BibTeX entry and citation info
82
+ ```bibtex
83
+ @misc{stan2023ldm3d,
84
+ title={LDM3D: Latent Diffusion Model for 3D},
85
+ author={Gabriela Ben Melech Stan and Diana Wofk and Scottie Fox and Alex Redden and Will Saxton and Jean Yu and Estelle Aflalo and Shao-Yen Tseng and Fabio Nonato and Matthias Muller and Vasudev Lal},
86
+ year={2023},
87
+ eprint={2305.10853},
88
+ archivePrefix={arXiv},
89
+ primaryClass={cs.CV}
90
+ }
91
+ ```