codestella
commited on
Commit
ยท
3c9f729
1
Parent(s):
e1b3138
README edit
Browse files- README.md +53 -27
- assets/chair.png +0 -0
- assets/comcom.jpeg +0 -0
- assets/drawing.png +0 -0
- assets/drum.png +0 -0
- assets/hotdog.png +0 -0
- assets/lego-14-occ-diet_.gif +0 -0
- assets/lego-14-occ-nerf_.gif +0 -0
- assets/lego-8-diet-image.png +0 -0
- assets/lego-8-diet.gif +0 -0
- assets/lego-8-nerf.gif +0 -0
- assets/materials-8-diet-image.png +0 -0
- assets/mic-8-diet.gif +0 -0
- assets/ship-8-diet-coarse.gif +0 -0
- assets/ship-8-nerf.gif +0 -0
README.md
CHANGED
@@ -2,13 +2,15 @@
|
|
2 |
|
3 |
<p align="center"><img width="450" alt="แแ
ณแแ
ณแ
แ
ตแซแแ
ฃแบ 2021-07-04 แแ
ฉแแ
ฎ 4 11 51" src="https://user-images.githubusercontent.com/77657524/126361638-4aad58e8-4efb-4fc5-bf78-f53d03799e1e.png"></p>
|
4 |
|
5 |
-
|
6 |
-
|
7 |
-
The
|
|
|
|
|
8 |
|
9 |
|
10 |
## ๐ค Hugging Face Hub Repo URL:
|
11 |
-
We will also upload our project on the Hugging Face Hub Repository.
|
12 |
[https://huggingface.co/flax-community/putting-nerf-on-a-diet/](https://huggingface.co/flax-community/putting-nerf-on-a-diet/)
|
13 |
|
14 |
Our JAX/Flax implementation currently supports:
|
@@ -46,6 +48,12 @@ Our JAX/Flax implementation currently supports:
|
|
46 |
</tbody>
|
47 |
</table>
|
48 |
|
|
|
|
|
|
|
|
|
|
|
|
|
49 |
## ๐ป Installation
|
50 |
|
51 |
```bash
|
@@ -65,15 +73,21 @@ pip install --upgrade jax jaxlib==0.1.57+cuda101 -f https://storage.googleapis.c
|
|
65 |
pip install flax transformer[flax]
|
66 |
```
|
67 |
|
68 |
-
## โฝ Dataset
|
69 |
Download the datasets from the [NeRF official Google Drive](https://drive.google.com/drive/folders/128yBriW1IG_3NJ5Rp7APSTZsJqdJdfc1).
|
70 |
Please download the `nerf_synthetic.zip` and unzip them
|
71 |
in the place you like. Let's assume they are placed under `/tmp/jaxnerf/data/`.
|
72 |
|
|
|
|
|
|
|
|
|
|
|
|
|
73 |
<p align="center"><img width="400" alt="แแ
ณแแ
ณแ
แ
ตแซแแ
ฃแบ 2021-07-04 แแ
ฉแแ
ฎ 4 11 51" src="https://user-images.githubusercontent.com/77657524/124376591-b312b780-dce2-11eb-80ad-9129d6f5eedb.png"></p>
|
74 |
|
75 |
Based on the principle
|
76 |
-
that โa bulldozer is a bulldozer from any perspectiveโ,
|
77 |
(DietNeRF cameras). This is possible because we compute a semantic consistency loss in a feature space capturing high-level
|
78 |
scene attributes, not in pixel space. We extract semantic representations of renderings using the CLIP Vision Transformer, then
|
79 |
maximize similarity with representations of ground-truth views. In
|
@@ -96,45 +110,51 @@ You can toggle the semantic loss by โuse_semantic_lossโ in configuration fil
|
|
96 |
|
97 |
## ๐ Expriment Result
|
98 |
|
99 |
-
|
100 |
### โ Rendered Rendering images by 8-shot learned Diet-NeRF
|
101 |
-
|
|
|
|
|
|
|
102 |
|
103 |
<p align="center">
|
104 |
<table>
|
105 |
<tr>
|
106 |
-
<td><img alt="" src="
|
107 |
-
|
108 |
-
|
|
|
|
|
|
|
|
|
109 |
<tr>
|
110 |
</table></p>
|
111 |
|
112 |
-
### โ
|
113 |
|
114 |
-
|
115 |
-
The
|
|
|
116 |
|
|
|
|
|
117 |
|
118 |
|
119 |
-
|
120 |
-
|
121 |
-
|
122 |
-
|
|
|
|
|
|
|
123 |
|
124 |
#### SHIP
|
125 |
<p align="center">
|
126 |
<table>
|
127 |
<tr>
|
128 |
-
<td><img alt="" src="
|
129 |
-
<td><img alt="" src="https://user-images.githubusercontent.com/26036843/126626308-1c63a1e1-5c87-42af-9d7d-6468bd345c6a.gif" width="300"/></td>
|
130 |
<tr>
|
131 |
</table></p>
|
132 |
|
133 |
-
## ๐คฉ Demo
|
134 |
-
|
135 |
-
You can check our Streamlit Space Demo on following site !
|
136 |
-
[https://huggingface.co/spaces/flax-community/DietNerf-Demo](https://huggingface.co/spaces/flax-community/DietNerf-Demo)
|
137 |
-
|
138 |
## ๐จโ๐งโ๐ฆ Our Teams
|
139 |
|
140 |
|
@@ -161,7 +181,7 @@ This project is based on โJAX-NeRFโ.
|
|
161 |
}
|
162 |
```
|
163 |
|
164 |
-
This project is based on โ
|
165 |
```
|
166 |
@misc{jain2021putting,
|
167 |
title={Putting NeRF on a Diet: Semantically Consistent Few-Shot View Synthesis},
|
@@ -182,8 +202,14 @@ This project is based on โJAX-NeRFโ.
|
|
182 |
Our Project is started in the HuggingFace X GoogleAI (JAX) Community Week Event.
|
183 |
https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/7104
|
184 |
|
185 |
-
Thank you for
|
186 |
Our team grows up with this community learning experience. It was wonderful time!
|
187 |
|
188 |
<p align="center"><img width="250" alt="แแ
ณแแ
ณแ
แ
ตแซแแ
ฃแบ 2021-07-04 แแ
ฉแแ
ฎ 4 11 51" src="https://user-images.githubusercontent.com/77657524/126369170-5664076c-ac99-4157-bc53-b91dfb7ed7e1.jpeg"></p>
|
189 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
|
3 |
<p align="center"><img width="450" alt="แแ
ณแแ
ณแ
แ
ตแซแแ
ฃแบ 2021-07-04 แแ
ฉแแ
ฎ 4 11 51" src="https://user-images.githubusercontent.com/77657524/126361638-4aad58e8-4efb-4fc5-bf78-f53d03799e1e.png"></p>
|
4 |
|
5 |
+
Welcome to Putting NeRF on a Diet Project!
|
6 |
+
This project is the Pytorch, JAX/Flax based code implementation of this paper [Putting NeRF on a Diet : Ajay Jain, Matthew Tancik, Pieter Abbeel, Arxiv : https://arxiv.org/abs/2104.00677]
|
7 |
+
The model generates the novel view synthesis redering (NeRF: Neural Radiances Field) with Fewshot learning scheme.
|
8 |
+
The semantic loss use the pre-trained CLIP Vision Transformer embedding. This information can give a 2D supervision for 3D.
|
9 |
+
The Diet NeRF result outperforms the original NeRF in 3D reconstruction and neural rendering with only few images.
|
10 |
|
11 |
|
12 |
## ๐ค Hugging Face Hub Repo URL:
|
13 |
+
We will also upload our project on the Hugging Face Hub Repository Also.
|
14 |
[https://huggingface.co/flax-community/putting-nerf-on-a-diet/](https://huggingface.co/flax-community/putting-nerf-on-a-diet/)
|
15 |
|
16 |
Our JAX/Flax implementation currently supports:
|
|
|
48 |
</tbody>
|
49 |
</table>
|
50 |
|
51 |
+
## ๐คฉ Demo
|
52 |
+
|
53 |
+
You can check our Streamlit Space demo on following site !
|
54 |
+
[https://huggingface.co/spaces/flax-community/DietNerf-Demo](https://huggingface.co/spaces/flax-community/DietNerf-Demo)
|
55 |
+
|
56 |
+
|
57 |
## ๐ป Installation
|
58 |
|
59 |
```bash
|
|
|
73 |
pip install flax transformer[flax]
|
74 |
```
|
75 |
|
76 |
+
## โฝ Dataset
|
77 |
Download the datasets from the [NeRF official Google Drive](https://drive.google.com/drive/folders/128yBriW1IG_3NJ5Rp7APSTZsJqdJdfc1).
|
78 |
Please download the `nerf_synthetic.zip` and unzip them
|
79 |
in the place you like. Let's assume they are placed under `/tmp/jaxnerf/data/`.
|
80 |
|
81 |
+
|
82 |
+
## ๐ Methods
|
83 |
+
|
84 |
+
You can check more detail explaination about DietNeRF on following **Notion Report**
|
85 |
+
* ๐๐ VEEEERY Detail DietNeRF Explaination Docs : (https://bit.ly/3x4FwcT)๐๐
|
86 |
+
|
87 |
<p align="center"><img width="400" alt="แแ
ณแแ
ณแ
แ
ตแซแแ
ฃแบ 2021-07-04 แแ
ฉแแ
ฎ 4 11 51" src="https://user-images.githubusercontent.com/77657524/124376591-b312b780-dce2-11eb-80ad-9129d6f5eedb.png"></p>
|
88 |
|
89 |
Based on the principle
|
90 |
+
that โa bulldozer is a bulldozer from any perspectiveโ, Our proposed DietNeRF supervises the radiance field from arbitrary poses
|
91 |
(DietNeRF cameras). This is possible because we compute a semantic consistency loss in a feature space capturing high-level
|
92 |
scene attributes, not in pixel space. We extract semantic representations of renderings using the CLIP Vision Transformer, then
|
93 |
maximize similarity with representations of ground-truth views. In
|
|
|
110 |
|
111 |
## ๐ Expriment Result
|
112 |
|
|
|
113 |
### โ Rendered Rendering images by 8-shot learned Diet-NeRF
|
114 |
+
|
115 |
+
DietNeRF has a strong capacity to generalise on novel and challenging views with EXTREMELY SMALL TRAINING SAMPLES!
|
116 |
+
|
117 |
+
### CHAIR / HOTDOG / DRUM / LEGO / MATERIALS
|
118 |
|
119 |
<p align="center">
|
120 |
<table>
|
121 |
<tr>
|
122 |
+
<td><img alt="" src="./assets/chair.png" width="400"/></td><td><img alt="" src="./assets/hotdog.png" width="400"/></td><td><img alt="" src="./assets/drum.png" width="400"/></td>
|
123 |
+
<tr>
|
124 |
+
</table></p>
|
125 |
+
<p align="center">
|
126 |
+
<table>
|
127 |
+
<tr>
|
128 |
+
<td><img alt="" src="./assets/lego-8-diet.gif" width="400"/></td><td><img alt="" src="./assets/mic-8-diet.gif" width="400"/></td>
|
129 |
<tr>
|
130 |
</table></p>
|
131 |
|
132 |
+
### โ Rendered GIF by occluded 14-shot learned NeRF and Diet-NeRF
|
133 |
|
134 |
+
We made aritificial occulusion on the right side of image (Only picked left side training poses).
|
135 |
+
The reconstruction quality can be compared with this experiment.
|
136 |
+
Diet NeRF shows better quailty than Original NeRF when It is occulused.
|
137 |
|
138 |
+
#### Training poses
|
139 |
+
<img width="1400" src="https://user-images.githubusercontent.com/26036843/126111980-4f332c87-a7f0-42e0-a355-8e77621bbca4.png">
|
140 |
|
141 |
|
142 |
+
#### LEGO
|
143 |
+
<p align="center">
|
144 |
+
<table>
|
145 |
+
<tr>
|
146 |
+
<td><img alt="" src="assets/lego-14-occ-diet_.gif" width="300"/></td><td><img alt="" src="assets/lego-14-occ-nerf_.gif" width="300"/></td>
|
147 |
+
<tr>
|
148 |
+
</table></p>
|
149 |
|
150 |
#### SHIP
|
151 |
<p align="center">
|
152 |
<table>
|
153 |
<tr>
|
154 |
+
<td><img alt="" src="./assets/ship-dietnerf.gif" width="300"/></td><td><img alt="" src="./assets/ship-nerf.gif" width="300"/></td>
|
|
|
155 |
<tr>
|
156 |
</table></p>
|
157 |
|
|
|
|
|
|
|
|
|
|
|
158 |
## ๐จโ๐งโ๐ฆ Our Teams
|
159 |
|
160 |
|
|
|
181 |
}
|
182 |
```
|
183 |
|
184 |
+
This project is based on โPutting NeRF on a Dietโ.
|
185 |
```
|
186 |
@misc{jain2021putting,
|
187 |
title={Putting NeRF on a Diet: Semantically Consistent Few-Shot View Synthesis},
|
|
|
202 |
Our Project is started in the HuggingFace X GoogleAI (JAX) Community Week Event.
|
203 |
https://discuss.huggingface.co/t/open-to-the-community-community-week-using-jax-flax-for-nlp-cv/7104
|
204 |
|
205 |
+
Thank you for our mentor Suraj and organizers in JAX/Flax Community Week!
|
206 |
Our team grows up with this community learning experience. It was wonderful time!
|
207 |
|
208 |
<p align="center"><img width="250" alt="แแ
ณแแ
ณแ
แ
ตแซแแ
ฃแบ 2021-07-04 แแ
ฉแแ
ฎ 4 11 51" src="https://user-images.githubusercontent.com/77657524/126369170-5664076c-ac99-4157-bc53-b91dfb7ed7e1.jpeg"></p>
|
209 |
|
210 |
+
Common Computer AI(https://comcom.ai/ko/) sponsored the multiple V100 GPUs for our project!
|
211 |
+
Thank you so much for your support!
|
212 |
+
<p align="center"><img width="250" alt="แแ
ณแแ
ณแ
แ
ตแซแแ
ฃแบ" src="./assets/comcom.jpeg"></p>
|
213 |
+
|
214 |
+
|
215 |
+
|
assets/chair.png
CHANGED
![]() |
![]() |
assets/comcom.jpeg
ADDED
![]() |
assets/drawing.png
CHANGED
![]() |
![]() |
assets/drum.png
CHANGED
![]() |
![]() |
assets/hotdog.png
CHANGED
![]() |
![]() |
assets/lego-14-occ-diet_.gif
ADDED
![]() |
assets/lego-14-occ-nerf_.gif
ADDED
![]() |
assets/lego-8-diet-image.png
ADDED
![]() |
assets/lego-8-diet.gif
ADDED
![]() |
assets/lego-8-nerf.gif
ADDED
![]() |
assets/materials-8-diet-image.png
ADDED
![]() |
assets/mic-8-diet.gif
ADDED
![]() |
assets/ship-8-diet-coarse.gif
ADDED
![]() |
assets/ship-8-nerf.gif
ADDED
![]() |