Hugging Face's logo

One-2-3-45
/
code

Text-to-3D
PyTorch
image-to-3d
Model card Files
xet
Community
2
Chao Xu commited on
Commit
14550cf
·
1 Parent(s): 1e5adf7

Update README

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. SparseNeuS_demo_v1/val.ipynb +0 -951
README.md CHANGED
@@ -11,14 +11,14 @@ license: mit
11
  <a style="display:inline-block; margin-left: .5em" href='https://github.com/One-2-3-45/One-2-3-45'><img src='https://img.shields.io/github/stars/One-2-3-45/One-2-3-45?style=social' /></a>
12
  </div>
13
 
14
- This inference model supports the demo for [One-2-3-45](http://One-2-3-45.com). Please check the [Space README](https://huggingface.co/spaces/One-2-3-45/One-2-3-45/blob/main/README.md) for local deployment of the demo.
15
 
16
  Try out our 🤗 Hugging Face Demo:
17
  <a target="_blank" href="https://huggingface.co/spaces/One-2-3-45/One-2-3-45">
18
  <img src="https://huggingface.co/datasets/huggingface/badges/raw/main/open-in-hf-spaces-sm.svg" alt="Open in HuggingFace"/>
19
  </a>
20
 
21
- Please refer to our [GitHub repo](https://github.com/One-2-3-45/One-2-3-45) for full code release.
22
 
23
  ## Citation
24
 
 
11
  <a style="display:inline-block; margin-left: .5em" href='https://github.com/One-2-3-45/One-2-3-45'><img src='https://img.shields.io/github/stars/One-2-3-45/One-2-3-45?style=social' /></a>
12
  </div>
13
 
14
+ This inference model supports the demo for [One-2-3-45](http://One-2-3-45.com).
15
 
16
  Try out our 🤗 Hugging Face Demo:
17
  <a target="_blank" href="https://huggingface.co/spaces/One-2-3-45/One-2-3-45">
18
  <img src="https://huggingface.co/datasets/huggingface/badges/raw/main/open-in-hf-spaces-sm.svg" alt="Open in HuggingFace"/>
19
  </a>
20
 
21
+ Please refer to our [GitHub repo](https://github.com/One-2-3-45/One-2-3-45) for full code release and local deployment.
22
 
23
  ## Citation
24
 
SparseNeuS_demo_v1/val.ipynb DELETED
@@ -1,951 +0,0 @@
1
- {
2
- "cells": [
3
- {
4
- "cell_type": "code",
5
- "execution_count": 3,
6
- "metadata": {},
7
- "outputs": [
8
- {
9
- "name": "stdout",
10
- "output_type": "stream",
11
- "text": [
12
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name gradio_tmp --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
13
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
14
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
15
- "\u001b[34mStore in: ../gradio_tmp\u001b[0m\n"
16
- ]
17
- },
18
- {
19
- "name": "stderr",
20
- "output_type": "stream",
21
- "text": [
22
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
23
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
24
- "[exp_runner_generic_blender_val.py:159 - __init__() ] Find checkpoint: ckpt_285000.pth\n",
25
- "[exp_runner_generic_blender_val.py:500 - load_checkpoint() ] End\n",
26
- "ic| self.iter_step: 285000, idx: -1\n",
27
- "[exp_runner_generic_blender_val.py:579 - export_mesh() ] Validate begin\n"
28
- ]
29
- },
30
- {
31
- "name": "stdout",
32
- "output_type": "stream",
33
- "text": [
34
- "time for getting data 0.0004603862762451172\n",
35
- "export mesh time: 5.274312734603882\n"
36
- ]
37
- },
38
- {
39
- "data": {
40
- "text/plain": [
41
- "0"
42
- ]
43
- },
44
- "execution_count": 3,
45
- "metadata": {},
46
- "output_type": "execute_result"
47
- }
48
- ],
49
- "source": [
50
- "import os \n",
51
- "\n",
52
- "dataset = 'gradio_tmp' # !!! the subfolder name in valpath for which you want to eval\n",
53
- "# os.system('pwd')\n",
54
- "\n",
55
- "bash_script = f'CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name {dataset} --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue'\n",
56
- "print(bash_script)\n",
57
- "os.system(bash_script)"
58
- ]
59
- },
60
- {
61
- "cell_type": "code",
62
- "execution_count": 3,
63
- "metadata": {},
64
- "outputs": [
65
- {
66
- "name": "stdout",
67
- "output_type": "stream",
68
- "text": [
69
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name eval_test/ebicycle2 --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
70
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
71
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
72
- "save mesh to: ../eval_test/ebicycle2\n",
73
- "\u001b[34mStore in: ../eval_test/ebicycle2\u001b[0m\n"
74
- ]
75
- },
76
- {
77
- "name": "stderr",
78
- "output_type": "stream",
79
- "text": [
80
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
81
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
82
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_180000.pth\n",
83
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
84
- "ic| self.iter_step: 180000, idx: -1\n",
85
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
86
- ]
87
- },
88
- {
89
- "name": "stdout",
90
- "output_type": "stream",
91
- "text": [
92
- "time for getting data 0.0004477500915527344\n",
93
- "export mesh time: 5.30308723449707\n"
94
- ]
95
- },
96
- {
97
- "data": {
98
- "text/plain": [
99
- "0"
100
- ]
101
- },
102
- "execution_count": 3,
103
- "metadata": {},
104
- "output_type": "execute_result"
105
- }
106
- ],
107
- "source": [
108
- "import os \n",
109
- "\n",
110
- "dataset = 'eval_test/ebicycle2' # !!! the subfolder name in valpath for which you want to eval\n",
111
- "# os.system('pwd')\n",
112
- "\n",
113
- "bash_script = f'CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name {dataset} --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue'\n",
114
- "print(bash_script)\n",
115
- "os.system(bash_script)"
116
- ]
117
- },
118
- {
119
- "cell_type": "code",
120
- "execution_count": 9,
121
- "metadata": {},
122
- "outputs": [
123
- {
124
- "name": "stderr",
125
- "output_type": "stream",
126
- "text": [
127
- " 0%| | 0/20 [00:00<?, ?it/s]"
128
- ]
129
- },
130
- {
131
- "name": "stdout",
132
- "output_type": "stream",
133
- "text": [
134
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/bigmac --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
135
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
136
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
137
- "save mesh to: ../mesh_pred/eval_real/bigmac\n",
138
- "\u001b[34mStore in: ../mesh_pred/eval_real/bigmac\u001b[0m\n"
139
- ]
140
- },
141
- {
142
- "name": "stderr",
143
- "output_type": "stream",
144
- "text": [
145
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
146
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
147
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
148
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
149
- "ic| self.iter_step: 190000, idx: -1\n",
150
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
151
- ]
152
- },
153
- {
154
- "name": "stdout",
155
- "output_type": "stream",
156
- "text": [
157
- "time for getting data 0.00046324729919433594\n",
158
- "export mesh time: 5.254480361938477\n"
159
- ]
160
- },
161
- {
162
- "name": "stderr",
163
- "output_type": "stream",
164
- "text": [
165
- " 5%|▌ | 1/20 [00:09<03:05, 9.78s/it]"
166
- ]
167
- },
168
- {
169
- "name": "stdout",
170
- "output_type": "stream",
171
- "text": [
172
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/dinosaur --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
173
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
174
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
175
- "save mesh to: ../mesh_pred/eval_real/dinosaur\n",
176
- "\u001b[34mStore in: ../mesh_pred/eval_real/dinosaur\u001b[0m\n"
177
- ]
178
- },
179
- {
180
- "name": "stderr",
181
- "output_type": "stream",
182
- "text": [
183
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
184
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
185
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
186
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
187
- "ic| self.iter_step: 190000, idx: -1\n",
188
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
189
- ]
190
- },
191
- {
192
- "name": "stdout",
193
- "output_type": "stream",
194
- "text": [
195
- "time for getting data 0.0004961490631103516\n",
196
- "export mesh time: 5.2382142543792725\n"
197
- ]
198
- },
199
- {
200
- "name": "stderr",
201
- "output_type": "stream",
202
- "text": [
203
- " 10%|█ | 2/20 [00:19<02:56, 9.79s/it]"
204
- ]
205
- },
206
- {
207
- "name": "stdout",
208
- "output_type": "stream",
209
- "text": [
210
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/goose_chef --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
211
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
212
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
213
- "save mesh to: ../mesh_pred/eval_real/goose_chef\n",
214
- "\u001b[34mStore in: ../mesh_pred/eval_real/goose_chef\u001b[0m\n"
215
- ]
216
- },
217
- {
218
- "name": "stderr",
219
- "output_type": "stream",
220
- "text": [
221
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
222
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
223
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
224
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
225
- "ic| self.iter_step: 190000, idx: -1\n",
226
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
227
- ]
228
- },
229
- {
230
- "name": "stdout",
231
- "output_type": "stream",
232
- "text": [
233
- "time for getting data 0.0004622936248779297\n",
234
- "export mesh time: 5.243659257888794\n"
235
- ]
236
- },
237
- {
238
- "name": "stderr",
239
- "output_type": "stream",
240
- "text": [
241
- " 15%|█▌ | 3/20 [00:29<02:46, 9.79s/it]"
242
- ]
243
- },
244
- {
245
- "name": "stdout",
246
- "output_type": "stream",
247
- "text": [
248
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/stool2 --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
249
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
250
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
251
- "save mesh to: ../mesh_pred/eval_real/stool2\n",
252
- "\u001b[34mStore in: ../mesh_pred/eval_real/stool2\u001b[0m\n"
253
- ]
254
- },
255
- {
256
- "name": "stderr",
257
- "output_type": "stream",
258
- "text": [
259
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
260
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
261
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
262
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
263
- "ic| self.iter_step: 190000, idx: -1\n",
264
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
265
- ]
266
- },
267
- {
268
- "name": "stdout",
269
- "output_type": "stream",
270
- "text": [
271
- "time for getting data 0.0004367828369140625\n",
272
- "export mesh time: 5.210400342941284\n"
273
- ]
274
- },
275
- {
276
- "name": "stderr",
277
- "output_type": "stream",
278
- "text": [
279
- " 20%|██ | 4/20 [00:39<02:36, 9.78s/it]"
280
- ]
281
- },
282
- {
283
- "name": "stdout",
284
- "output_type": "stream",
285
- "text": [
286
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/nuts --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
287
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
288
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
289
- "save mesh to: ../mesh_pred/eval_real/nuts\n",
290
- "\u001b[34mStore in: ../mesh_pred/eval_real/nuts\u001b[0m\n"
291
- ]
292
- },
293
- {
294
- "name": "stderr",
295
- "output_type": "stream",
296
- "text": [
297
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
298
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
299
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
300
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
301
- "ic| self.iter_step: 190000, idx: -1\n",
302
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
303
- ]
304
- },
305
- {
306
- "name": "stdout",
307
- "output_type": "stream",
308
- "text": [
309
- "time for getting data 0.00047707557678222656\n",
310
- "export mesh time: 5.257585763931274\n"
311
- ]
312
- },
313
- {
314
- "name": "stderr",
315
- "output_type": "stream",
316
- "text": [
317
- " 25%|██▌ | 5/20 [00:48<02:27, 9.80s/it]"
318
- ]
319
- },
320
- {
321
- "name": "stdout",
322
- "output_type": "stream",
323
- "text": [
324
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/yellow_duck --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
325
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
326
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
327
- "save mesh to: ../mesh_pred/eval_real/yellow_duck\n",
328
- "\u001b[34mStore in: ../mesh_pred/eval_real/yellow_duck\u001b[0m\n"
329
- ]
330
- },
331
- {
332
- "name": "stderr",
333
- "output_type": "stream",
334
- "text": [
335
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
336
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
337
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
338
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
339
- "ic| self.iter_step: 190000, idx: -1\n",
340
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
341
- ]
342
- },
343
- {
344
- "name": "stdout",
345
- "output_type": "stream",
346
- "text": [
347
- "time for getting data 0.0004456043243408203\n",
348
- "export mesh time: 5.2359161376953125\n"
349
- ]
350
- },
351
- {
352
- "name": "stderr",
353
- "output_type": "stream",
354
- "text": [
355
- " 30%|███ | 6/20 [00:58<02:17, 9.80s/it]"
356
- ]
357
- },
358
- {
359
- "name": "stdout",
360
- "output_type": "stream",
361
- "text": [
362
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/pineapple_bottle --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
363
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
364
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
365
- "save mesh to: ../mesh_pred/eval_real/pineapple_bottle\n",
366
- "\u001b[34mStore in: ../mesh_pred/eval_real/pineapple_bottle\u001b[0m\n"
367
- ]
368
- },
369
- {
370
- "name": "stderr",
371
- "output_type": "stream",
372
- "text": [
373
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
374
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
375
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
376
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
377
- "ic| self.iter_step: 190000, idx: -1\n",
378
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
379
- ]
380
- },
381
- {
382
- "name": "stdout",
383
- "output_type": "stream",
384
- "text": [
385
- "time for getting data 0.0004482269287109375\n",
386
- "export mesh time: 5.260643243789673\n"
387
- ]
388
- },
389
- {
390
- "name": "stderr",
391
- "output_type": "stream",
392
- "text": [
393
- " 35%|███▌ | 7/20 [01:08<02:07, 9.82s/it]"
394
- ]
395
- },
396
- {
397
- "name": "stdout",
398
- "output_type": "stream",
399
- "text": [
400
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/pancake --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
401
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
402
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
403
- "save mesh to: ../mesh_pred/eval_real/pancake\n",
404
- "\u001b[34mStore in: ../mesh_pred/eval_real/pancake\u001b[0m\n"
405
- ]
406
- },
407
- {
408
- "name": "stderr",
409
- "output_type": "stream",
410
- "text": [
411
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
412
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
413
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
414
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
415
- "ic| self.iter_step: 190000, idx: -1\n",
416
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
417
- ]
418
- },
419
- {
420
- "name": "stdout",
421
- "output_type": "stream",
422
- "text": [
423
- "time for getting data 0.00047278404235839844\n",
424
- "export mesh time: 5.221261739730835\n"
425
- ]
426
- },
427
- {
428
- "name": "stderr",
429
- "output_type": "stream",
430
- "text": [
431
- " 40%|████ | 8/20 [01:18<01:57, 9.83s/it]"
432
- ]
433
- },
434
- {
435
- "name": "stdout",
436
- "output_type": "stream",
437
- "text": [
438
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/hydrant --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
439
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
440
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
441
- "save mesh to: ../mesh_pred/eval_real/hydrant\n",
442
- "\u001b[34mStore in: ../mesh_pred/eval_real/hydrant\u001b[0m\n"
443
- ]
444
- },
445
- {
446
- "name": "stderr",
447
- "output_type": "stream",
448
- "text": [
449
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
450
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
451
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
452
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
453
- "ic| self.iter_step: 190000, idx: -1\n",
454
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
455
- ]
456
- },
457
- {
458
- "name": "stdout",
459
- "output_type": "stream",
460
- "text": [
461
- "time for getting data 0.00044608116149902344\n",
462
- "export mesh time: 5.220775604248047\n"
463
- ]
464
- },
465
- {
466
- "name": "stderr",
467
- "output_type": "stream",
468
- "text": [
469
- " 45%|████▌ | 9/20 [01:28<01:47, 9.80s/it]"
470
- ]
471
- },
472
- {
473
- "name": "stdout",
474
- "output_type": "stream",
475
- "text": [
476
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/oreo_drink --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
477
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
478
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
479
- "save mesh to: ../mesh_pred/eval_real/oreo_drink\n",
480
- "\u001b[34mStore in: ../mesh_pred/eval_real/oreo_drink\u001b[0m\n"
481
- ]
482
- },
483
- {
484
- "name": "stderr",
485
- "output_type": "stream",
486
- "text": [
487
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
488
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
489
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
490
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
491
- "ic| self.iter_step: 190000, idx: -1\n",
492
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
493
- ]
494
- },
495
- {
496
- "name": "stdout",
497
- "output_type": "stream",
498
- "text": [
499
- "time for getting data 0.00043773651123046875\n",
500
- "export mesh time: 5.22020149230957\n"
501
- ]
502
- },
503
- {
504
- "name": "stderr",
505
- "output_type": "stream",
506
- "text": [
507
- " 50%|█████ | 10/20 [01:37<01:37, 9.79s/it]"
508
- ]
509
- },
510
- {
511
- "name": "stdout",
512
- "output_type": "stream",
513
- "text": [
514
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/scissor --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
515
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
516
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
517
- "save mesh to: ../mesh_pred/eval_real/scissor\n",
518
- "\u001b[34mStore in: ../mesh_pred/eval_real/scissor\u001b[0m\n"
519
- ]
520
- },
521
- {
522
- "name": "stderr",
523
- "output_type": "stream",
524
- "text": [
525
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
526
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
527
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
528
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
529
- "ic| self.iter_step: 190000, idx: -1\n",
530
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
531
- ]
532
- },
533
- {
534
- "name": "stdout",
535
- "output_type": "stream",
536
- "text": [
537
- "time for getting data 0.0004305839538574219\n",
538
- "export mesh time: 5.205048322677612\n"
539
- ]
540
- },
541
- {
542
- "name": "stderr",
543
- "output_type": "stream",
544
- "text": [
545
- " 55%|█████▌ | 11/20 [01:47<01:28, 9.78s/it]"
546
- ]
547
- },
548
- {
549
- "name": "stdout",
550
- "output_type": "stream",
551
- "text": [
552
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/chocolatecake --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
553
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
554
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
555
- "save mesh to: ../mesh_pred/eval_real/chocolatecake\n",
556
- "\u001b[34mStore in: ../mesh_pred/eval_real/chocolatecake\u001b[0m\n"
557
- ]
558
- },
559
- {
560
- "name": "stderr",
561
- "output_type": "stream",
562
- "text": [
563
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
564
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
565
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
566
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
567
- "ic| self.iter_step: 190000, idx: -1\n",
568
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
569
- ]
570
- },
571
- {
572
- "name": "stdout",
573
- "output_type": "stream",
574
- "text": [
575
- "time for getting data 0.00043392181396484375\n",
576
- "export mesh time: 5.239720344543457\n"
577
- ]
578
- },
579
- {
580
- "name": "stderr",
581
- "output_type": "stream",
582
- "text": [
583
- " 60%|██████ | 12/20 [01:57<01:18, 9.79s/it]"
584
- ]
585
- },
586
- {
587
- "name": "stdout",
588
- "output_type": "stream",
589
- "text": [
590
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/mario --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
591
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
592
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
593
- "save mesh to: ../mesh_pred/eval_real/mario\n",
594
- "\u001b[34mStore in: ../mesh_pred/eval_real/mario\u001b[0m\n"
595
- ]
596
- },
597
- {
598
- "name": "stderr",
599
- "output_type": "stream",
600
- "text": [
601
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
602
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
603
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
604
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
605
- "ic| self.iter_step: 190000, idx: -1\n",
606
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
607
- ]
608
- },
609
- {
610
- "name": "stdout",
611
- "output_type": "stream",
612
- "text": [
613
- "time for getting data 0.0004582405090332031\n",
614
- "export mesh time: 5.255834102630615\n"
615
- ]
616
- },
617
- {
618
- "name": "stderr",
619
- "output_type": "stream",
620
- "text": [
621
- " 65%|██████▌ | 13/20 [02:07<01:08, 9.79s/it]"
622
- ]
623
- },
624
- {
625
- "name": "stdout",
626
- "output_type": "stream",
627
- "text": [
628
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/broccoli --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
629
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
630
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
631
- "save mesh to: ../mesh_pred/eval_real/broccoli\n",
632
- "\u001b[34mStore in: ../mesh_pred/eval_real/broccoli\u001b[0m\n"
633
- ]
634
- },
635
- {
636
- "name": "stderr",
637
- "output_type": "stream",
638
- "text": [
639
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
640
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
641
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
642
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
643
- "ic| self.iter_step: 190000, idx: -1\n",
644
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
645
- ]
646
- },
647
- {
648
- "name": "stdout",
649
- "output_type": "stream",
650
- "text": [
651
- "time for getting data 0.0005385875701904297\n",
652
- "export mesh time: 5.311108112335205\n"
653
- ]
654
- },
655
- {
656
- "name": "stderr",
657
- "output_type": "stream",
658
- "text": [
659
- " 70%|███████ | 14/20 [02:17<00:58, 9.83s/it]"
660
- ]
661
- },
662
- {
663
- "name": "stdout",
664
- "output_type": "stream",
665
- "text": [
666
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/chair3 --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
667
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
668
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
669
- "save mesh to: ../mesh_pred/eval_real/chair3\n",
670
- "\u001b[34mStore in: ../mesh_pred/eval_real/chair3\u001b[0m\n"
671
- ]
672
- },
673
- {
674
- "name": "stderr",
675
- "output_type": "stream",
676
- "text": [
677
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
678
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
679
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
680
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
681
- "ic| self.iter_step: 190000, idx: -1\n",
682
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
683
- ]
684
- },
685
- {
686
- "name": "stdout",
687
- "output_type": "stream",
688
- "text": [
689
- "time for getting data 0.0004534721374511719\n",
690
- "export mesh time: 5.227793216705322\n"
691
- ]
692
- },
693
- {
694
- "name": "stderr",
695
- "output_type": "stream",
696
- "text": [
697
- " 75%|███████▌ | 15/20 [02:27<00:49, 9.82s/it]"
698
- ]
699
- },
700
- {
701
- "name": "stdout",
702
- "output_type": "stream",
703
- "text": [
704
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/lysol --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
705
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
706
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
707
- "save mesh to: ../mesh_pred/eval_real/lysol\n",
708
- "\u001b[34mStore in: ../mesh_pred/eval_real/lysol\u001b[0m\n"
709
- ]
710
- },
711
- {
712
- "name": "stderr",
713
- "output_type": "stream",
714
- "text": [
715
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
716
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
717
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
718
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
719
- "ic| self.iter_step: 190000, idx: -1\n",
720
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
721
- ]
722
- },
723
- {
724
- "name": "stdout",
725
- "output_type": "stream",
726
- "text": [
727
- "time for getting data 0.00045418739318847656\n",
728
- "export mesh time: 5.2634806632995605\n"
729
- ]
730
- },
731
- {
732
- "name": "stderr",
733
- "output_type": "stream",
734
- "text": [
735
- " 80%|████████ | 16/20 [02:36<00:39, 9.82s/it]"
736
- ]
737
- },
738
- {
739
- "name": "stdout",
740
- "output_type": "stream",
741
- "text": [
742
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/clock2 --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
743
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
744
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
745
- "save mesh to: ../mesh_pred/eval_real/clock2\n",
746
- "\u001b[34mStore in: ../mesh_pred/eval_real/clock2\u001b[0m\n"
747
- ]
748
- },
749
- {
750
- "name": "stderr",
751
- "output_type": "stream",
752
- "text": [
753
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
754
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
755
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
756
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
757
- "ic| self.iter_step: 190000, idx: -1\n",
758
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
759
- ]
760
- },
761
- {
762
- "name": "stdout",
763
- "output_type": "stream",
764
- "text": [
765
- "time for getting data 0.00043654441833496094\n",
766
- "export mesh time: 5.26728630065918\n"
767
- ]
768
- },
769
- {
770
- "name": "stderr",
771
- "output_type": "stream",
772
- "text": [
773
- " 85%|████████▌ | 17/20 [02:46<00:29, 9.84s/it]"
774
- ]
775
- },
776
- {
777
- "name": "stdout",
778
- "output_type": "stream",
779
- "text": [
780
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/downy --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
781
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
782
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
783
- "save mesh to: ../mesh_pred/eval_real/downy\n",
784
- "\u001b[34mStore in: ../mesh_pred/eval_real/downy\u001b[0m\n"
785
- ]
786
- },
787
- {
788
- "name": "stderr",
789
- "output_type": "stream",
790
- "text": [
791
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
792
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
793
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
794
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
795
- "ic| self.iter_step: 190000, idx: -1\n",
796
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
797
- ]
798
- },
799
- {
800
- "name": "stdout",
801
- "output_type": "stream",
802
- "text": [
803
- "time for getting data 0.0004367828369140625\n",
804
- "export mesh time: 5.207308769226074\n"
805
- ]
806
- },
807
- {
808
- "name": "stderr",
809
- "output_type": "stream",
810
- "text": [
811
- " 90%|█████████ | 18/20 [02:56<00:19, 9.82s/it]"
812
- ]
813
- },
814
- {
815
- "name": "stdout",
816
- "output_type": "stream",
817
- "text": [
818
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/ebicycle2 --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
819
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
820
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
821
- "save mesh to: ../mesh_pred/eval_real/ebicycle2\n",
822
- "\u001b[34mStore in: ../mesh_pred/eval_real/ebicycle2\u001b[0m\n"
823
- ]
824
- },
825
- {
826
- "name": "stderr",
827
- "output_type": "stream",
828
- "text": [
829
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
830
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
831
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
832
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
833
- "ic| self.iter_step: 190000, idx: -1\n",
834
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
835
- ]
836
- },
837
- {
838
- "name": "stdout",
839
- "output_type": "stream",
840
- "text": [
841
- "time for getting data 0.00042724609375\n",
842
- "export mesh time: 5.277446985244751\n"
843
- ]
844
- },
845
- {
846
- "name": "stderr",
847
- "output_type": "stream",
848
- "text": [
849
- " 95%|█████████▌| 19/20 [03:06<00:09, 9.84s/it]"
850
- ]
851
- },
852
- {
853
- "name": "stdout",
854
- "output_type": "stream",
855
- "text": [
856
- "CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name mesh_pred/eval_real/strawberrycake --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue\n",
857
- "\u001b[31mdetected 1 GPUs\u001b[0m\n",
858
- "\u001b[33mbase_exp_dir: exp/lod0\u001b[0m\n",
859
- "save mesh to: ../mesh_pred/eval_real/strawberrycake\n",
860
- "\u001b[34mStore in: ../mesh_pred/eval_real/strawberrycake\u001b[0m\n"
861
- ]
862
- },
863
- {
864
- "name": "stderr",
865
- "output_type": "stream",
866
- "text": [
867
- "/home/chao/anaconda3/envs/gradio/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)\n",
868
- " return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]\n",
869
- "[exp_runner_generic_blender_val.py:160 - __init__() ] Find checkpoint: ckpt_190000.pth\n",
870
- "[exp_runner_generic_blender_val.py:501 - load_checkpoint() ] End\n",
871
- "ic| self.iter_step: 190000, idx: -1\n",
872
- "[exp_runner_generic_blender_val.py:580 - export_mesh() ] Validate begin\n"
873
- ]
874
- },
875
- {
876
- "name": "stdout",
877
- "output_type": "stream",
878
- "text": [
879
- "time for getting data 0.0004417896270751953\n",
880
- "export mesh time: 5.276712417602539\n"
881
- ]
882
- },
883
- {
884
- "name": "stderr",
885
- "output_type": "stream",
886
- "text": [
887
- "100%|██████████| 20/20 [03:16<00:00, 9.81s/it]\n"
888
- ]
889
- }
890
- ],
891
- "source": [
892
- "import os \n",
893
- "from tqdm import tqdm\n",
894
- "base_dir = \"mesh_pred/eval_real\"\n",
895
- "for nickname in tqdm(os.listdir(\"../\"+base_dir)):\n",
896
- " dataset = os.path.join(base_dir, nickname)\n",
897
- " bash_script = f'CUDA_VISIBLE_DEVICES=3 python exp_runner_generic_blender_val.py --specific_dataset_name {dataset} --mode export_mesh --conf confs/one2345_lod0_val_demo.conf --is_continue'\n",
898
- " print(bash_script)\n",
899
- " os.system(bash_script)"
900
- ]
901
- },
902
- {
903
- "cell_type": "code",
904
- "execution_count": 10,
905
- "metadata": {},
906
- "outputs": [
907
- {
908
- "name": "stderr",
909
- "output_type": "stream",
910
- "text": [
911
- "100%|██████████| 20/20 [00:00<00:00, 1310.90it/s]\n"
912
- ]
913
- }
914
- ],
915
- "source": [
916
- "import os \n",
917
- "from tqdm import tqdm\n",
918
- "from shutil import copyfile\n",
919
- "base_dir = \"mesh_pred/eval_real\"\n",
920
- "out_dir = f\"../{base_dir}_mesh\"\n",
921
- "os.makedirs(out_dir, exist_ok=True)\n",
922
- "for nickname in tqdm(os.listdir(\"../\"+base_dir)):\n",
923
- " mesh_path = os.path.join(\"../\", base_dir, nickname, \"meshes_val_bg/lod0/mesh_00190000_gradio_lod0.ply\")\n",
924
- " target_path = os.path.join(out_dir, nickname+\".ply\")\n",
925
- " copyfile(mesh_path, target_path)"
926
- ]
927
- }
928
- ],
929
- "metadata": {
930
- "kernelspec": {
931
- "display_name": "base",
932
- "language": "python",
933
- "name": "python3"
934
- },
935
- "language_info": {
936
- "codemirror_mode": {
937
- "name": "ipython",
938
- "version": 3
939
- },
940
- "file_extension": ".py",
941
- "mimetype": "text/x-python",
942
- "name": "python",
943
- "nbconvert_exporter": "python",
944
- "pygments_lexer": "ipython3",
945
- "version": "3.10.11"
946
- },
947
- "orig_nbformat": 4
948
- },
949
- "nbformat": 4,
950
- "nbformat_minor": 2
951
- }