estellea commited on
Commit
f0b0083
1 Parent(s): c278514

updated readme with description

Browse files
Files changed (1) hide show
  1. README.md +75 -1
README.md CHANGED
@@ -4,4 +4,78 @@ datasets:
4
  language:
5
  - en
6
  pipeline_tag: text-to-image
7
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  language:
5
  - en
6
  pipeline_tag: text-to-image
7
+ ---
8
+
9
+
10
+
11
+ # LDM3D model
12
+
13
+ The LDM3D model was proposed in "LDM3D: Latent Diffusion Model for 3D" 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.
14
+ It was introduced in [this paper](https://arxiv.org/abs/2305.10853.pdf)
15
+
16
+ LDM3D got accepted to [CVPRW'23](https://cvpr2023.thecvf.com/).
17
+
18
+ ## Model description
19
+
20
+ The abstract from the paper is the following:
21
+ 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
22
+
23
+ ## Intended uses & limitations(TODO)
24
+
25
+ You can use the model to generate RGB and depth images from text prompt.
26
+ 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)
27
+
28
+ ### How to use
29
+
30
+ Here is how to use this model:
31
+
32
+ ```python
33
+
34
+ from diffusers import StableDiffusionLDM3DPipeline
35
+
36
+ pipe_ldm3d = StableDiffusionLDM3DPipeline.from_pretrained("LDM3D/ldm3d-v1")
37
+ pipe_ldm3d.to("cuda")
38
+
39
+ prompt ="A picture of some lemons on a table"
40
+ name = "lemons"
41
+
42
+ rgb_image, depth_image = pipe_ldm3d(prompt).images
43
+ rgb_image[0].save(name+"_ldm3d_rgb.jpg")
44
+ depth_image[0].save(name+"_ldm3d_depth.png")
45
+ ```
46
+ ### Limitations and bias
47
+
48
+ TODO
49
+
50
+ ## Training data
51
+
52
+ The LDM3D model was finetuned on a subset of the LAION-400M dataset, a large-scale image-caption dataset that contains over 400 million image-caption pairs
53
+
54
+
55
+ ## Training procedure
56
+
57
+ 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.
58
+
59
+ ### Preprocessing
60
+
61
+ TODO
62
+
63
+ ### Compute Infrastructure
64
+
65
+ All training runs reported in this work are conducted on an Intel AI supercomputing cluster comprising of Intel Xeon processors and Intel Habana Gaudi AI accelerators. The LDM3D model training run is scaled out to 16 accelerators (Gaudis) on the corpus of 9,600 tupples (text caption, RGB image, depth map). The KL-autoencoder used in our LDM3D model was trained on Nvidia A6000 GPUs.
66
+
67
+ ## Evaluation results
68
+
69
+ Please refer to Table 1 and 2 of the [paper](https://arxiv.org/pdf/2305.10853.pdf)
70
+
71
+ ### BibTeX entry and citation info
72
+ ```bibtex
73
+ @misc{stan2023ldm3d,
74
+ title={LDM3D: Latent Diffusion Model for 3D},
75
+ 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},
76
+ year={2023},
77
+ eprint={2305.10853},
78
+ archivePrefix={arXiv},
79
+ primaryClass={cs.CV}
80
+ }
81
+ ```