updated readme with description
Browse files
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 |
+
```
|