diff --git a/.gitignore b/.gitignore index 763b71bc0d6395006143a864832e09801fd4fb11..895f30fa38c67c122c8479a4461350153dacff05 100644 --- a/.gitignore +++ b/.gitignore @@ -162,4 +162,5 @@ training core* tmp -logs \ No newline at end of file +logs +paper_figures \ No newline at end of file diff --git a/app.py b/app.py index 5d7276bb781927bf98dc3c122dacc186a55161aa..45acb9dc04b6d88e6b3604711334af322a1bfa61 100644 --- a/app.py +++ b/app.py @@ -1,4 +1,4 @@ -import spaces # RuntimeError: CUDA has been initialized before importing the `spaces` package +import spaces import os import torch import sys @@ -290,7 +290,9 @@ def main(args_1, args_2): Locally, on an NVIDIA A100/A10 GPU, each image-conditioned diffusion generation can be done within 20 seconds (time varies due to the adaptive-step ODE solver used in flow-mathcing.) Upload an image of an object or click on one of the provided examples to see how the GaussianAnything works. - The 3D viewer will render a .glb point cloud exported from the centers of the surfel Gaussians, and an integrated TSDF mesh. + The 3D viewer will render a .glb point cloud exported from the centers of the surfel Gaussians, and an integrated TSDF mesh. + Besides, you can find the intermediate stage-1 point cloud in the Tab (Stage-1 Output). + For best results run the demo locally and render locally - to do so, clone the [main repository](https://github.com/NIRVANALAN/GaussianAnything). """ ) @@ -356,7 +358,7 @@ def main(args_1, args_2): with gr.Row(): with gr.Tab("Stage-2 Output"): with gr.Column(): - output_video = gr.Video(value=None, width=512, label="Rendered Video (2 LoDs)", autoplay=True, loop=True) + output_video = gr.Video(value=None, width=384, label="Rendered Video (2 LoDs)", autoplay=True, loop=True) # output_video = gr.Video(value=None, width=256, label="Rendered Video", autoplay=True) output_gs = gr.Model3D( height=256, @@ -386,11 +388,9 @@ def main(args_1, args_2): """ ## Comments: 1. The sampling time varies since ODE-based sampling method (dopri5 by default) has adaptive internal step, and reducing sampling steps may not reduce the overal sampling time. Sampling steps=250 is the emperical value that works well in most cases. - 2. The 3D viewer shows a colored .glb mesh extracted from volumetric tri-plane, and may differ slightly with the volume rendering result. + 2. The 3D viewer shows a colored .glb point cloud extracted from 2DGS center and xyz, as well as a TSDF integrated mesh from multi-view RGBD images. 3. If you find your result unsatisfying, tune the CFG scale and change the random seed. Usually slightly increase the CFG value can lead to better performance. - 3. Known limitations include: - - Texture details missing: since our VAE is trained on 192x192 resolution due the the resource constraints, the texture details generated by the final 3D-LDM may be blurry. We will keep improving the performance in the future. - 4. Regarding reconstruction performance, our model is slightly inferior to state-of-the-art multi-view LRM-based method (e.g. InstantMesh), but offers much better diversity, flexibility and editing potential due to the intrinsic nature of diffusion model. + 4. Regarding image-to-3D reconstruction performance, our model is slightly inferior to state-of-the-art multi-view LRM-based method (e.g. InstantMesh), but offers much better diversity, flexibility and editing potential due to the intrinsic nature of diffusion model. ## How does it work? diff --git a/cldm/__pycache__/cldm.cpython-310.pyc b/cldm/__pycache__/cldm.cpython-310.pyc deleted file mode 100644 index 416350827cc2229e320fdf877156adca3e051543..0000000000000000000000000000000000000000 Binary files a/cldm/__pycache__/cldm.cpython-310.pyc and /dev/null differ diff --git a/cldm/__pycache__/cldm.cpython-39.pyc b/cldm/__pycache__/cldm.cpython-39.pyc deleted file mode 100644 index 70a61eda8d5c67b153dabc7dadd6614f230d8114..0000000000000000000000000000000000000000 Binary files a/cldm/__pycache__/cldm.cpython-39.pyc and /dev/null differ diff --git a/datasets/__pycache__/__init__.cpython-39.pyc b/datasets/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 68ca17d43f9a3f027f8f98e198291fe390899fe0..0000000000000000000000000000000000000000 Binary files a/datasets/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/datasets/__pycache__/eg3d_dataset.cpython-39.pyc b/datasets/__pycache__/eg3d_dataset.cpython-39.pyc deleted file mode 100644 index 68ceab7814f1baafd817c7ae006da82e87624596..0000000000000000000000000000000000000000 Binary files a/datasets/__pycache__/eg3d_dataset.cpython-39.pyc and /dev/null differ diff --git a/datasets/__pycache__/g_buffer_objaverse.cpython-39.pyc b/datasets/__pycache__/g_buffer_objaverse.cpython-39.pyc deleted file mode 100644 index 8d3818033ceb5bdcb878afa21d7a2050c5b8f4b0..0000000000000000000000000000000000000000 Binary files a/datasets/__pycache__/g_buffer_objaverse.cpython-39.pyc and /dev/null differ diff --git a/datasets/__pycache__/shapenet.cpython-39.pyc b/datasets/__pycache__/shapenet.cpython-39.pyc deleted file mode 100644 index 32039e600121e192e12adfe4af4c29568f0a0a89..0000000000000000000000000000000000000000 Binary files a/datasets/__pycache__/shapenet.cpython-39.pyc and /dev/null differ diff --git a/dit/__pycache__/__init__.cpython-310.pyc b/dit/__pycache__/__init__.cpython-310.pyc deleted file mode 100644 index c086749465eb3cd6305018c46b9e88158e953a92..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/__init__.cpython-310.pyc and /dev/null differ diff --git a/dit/__pycache__/__init__.cpython-39.pyc b/dit/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index b274a4a751e4bb7e1ffba0d0fbb382de56c1dc32..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/dit/__pycache__/dit_decoder.cpython-39.pyc b/dit/__pycache__/dit_decoder.cpython-39.pyc deleted file mode 100644 index f1d3d0d3b551e503def90a19bed5fcf54fad7fc9..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/dit_decoder.cpython-39.pyc and /dev/null differ diff --git a/dit/__pycache__/dit_i23d.cpython-39.pyc b/dit/__pycache__/dit_i23d.cpython-39.pyc deleted file mode 100644 index 535b4b5b049de3f5ffb7203ac2f4b669e03bc8ad..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/dit_i23d.cpython-39.pyc and /dev/null differ diff --git a/dit/__pycache__/dit_models.cpython-39.pyc b/dit/__pycache__/dit_models.cpython-39.pyc deleted file mode 100644 index 951c6a0ef553ca294a17ac44839156e03ba345e2..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/dit_models.cpython-39.pyc and /dev/null differ diff --git a/dit/__pycache__/dit_models_xformers.cpython-310.pyc b/dit/__pycache__/dit_models_xformers.cpython-310.pyc deleted file mode 100644 index bb8ad85b31d12667c36f3c3dca83b7155c848e3d..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/dit_models_xformers.cpython-310.pyc and /dev/null differ diff --git a/dit/__pycache__/dit_models_xformers.cpython-39.pyc b/dit/__pycache__/dit_models_xformers.cpython-39.pyc deleted file mode 100644 index c182cc582ba716b5a5e1c2197f50b32df9ade873..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/dit_models_xformers.cpython-39.pyc and /dev/null differ diff --git a/dit/__pycache__/dit_trilatent.cpython-310.pyc b/dit/__pycache__/dit_trilatent.cpython-310.pyc deleted file mode 100644 index d6cb16015fea69da0eaead6077f2b675e3c9adc4..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/dit_trilatent.cpython-310.pyc and /dev/null differ diff --git a/dit/__pycache__/dit_trilatent.cpython-39.pyc b/dit/__pycache__/dit_trilatent.cpython-39.pyc deleted file mode 100644 index 519a43ffb693ca52498047b5c72c69ecb3cfdac3..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/dit_trilatent.cpython-39.pyc and /dev/null differ diff --git a/dit/__pycache__/norm.cpython-310.pyc b/dit/__pycache__/norm.cpython-310.pyc deleted file mode 100644 index a88c3400e4afd7ecf3df1959124447adc41bd708..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/norm.cpython-310.pyc and /dev/null differ diff --git a/dit/__pycache__/norm.cpython-39.pyc b/dit/__pycache__/norm.cpython-39.pyc deleted file mode 100644 index 7b5009b8d7ff5093d0fee88fe02054a8f7755a94..0000000000000000000000000000000000000000 Binary files a/dit/__pycache__/norm.cpython-39.pyc and /dev/null differ diff --git a/dnnlib/__pycache__/__init__.cpython-310.pyc b/dnnlib/__pycache__/__init__.cpython-310.pyc deleted file mode 100644 index ba6aedbb18cdeab74e8414837d640642e33f7c23..0000000000000000000000000000000000000000 Binary files a/dnnlib/__pycache__/__init__.cpython-310.pyc and /dev/null differ diff --git a/dnnlib/__pycache__/__init__.cpython-39.pyc b/dnnlib/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index f36aa96f44ea1d8cd73a23ad8135eb0fdb1b982e..0000000000000000000000000000000000000000 Binary files a/dnnlib/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/dnnlib/__pycache__/util.cpython-310.pyc b/dnnlib/__pycache__/util.cpython-310.pyc deleted file mode 100644 index 98284bb3ab6c918240a2277cb9adfd62ee8a7569..0000000000000000000000000000000000000000 Binary files a/dnnlib/__pycache__/util.cpython-310.pyc and /dev/null differ diff --git a/dnnlib/__pycache__/util.cpython-39.pyc b/dnnlib/__pycache__/util.cpython-39.pyc deleted file mode 100644 index 867125acf0aa14f5454d57d8c714ddbfb23e85a0..0000000000000000000000000000000000000000 Binary files a/dnnlib/__pycache__/util.cpython-39.pyc and /dev/null differ diff --git a/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/feature_extractor.cpython-39.pyc b/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/feature_extractor.cpython-39.pyc deleted file mode 100644 index 53ba7f74caa240b0958c88d3b0972d2a814113cc..0000000000000000000000000000000000000000 Binary files a/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/feature_extractor.cpython-39.pyc and /dev/null differ diff --git a/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/npz_stream.cpython-39.pyc b/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/npz_stream.cpython-39.pyc deleted file mode 100644 index 6e648ea5b507fb2f2f1c9387073a4833c076349b..0000000000000000000000000000000000000000 Binary files a/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/npz_stream.cpython-39.pyc and /dev/null differ diff --git a/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/pointnet2_cls_ssg.cpython-39.pyc b/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/pointnet2_cls_ssg.cpython-39.pyc deleted file mode 100644 index b5937f9c49fa6b4d2ca12f68efa31e82b236a9c1..0000000000000000000000000000000000000000 Binary files a/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/pointnet2_cls_ssg.cpython-39.pyc and /dev/null differ diff --git a/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/pointnet2_utils.cpython-39.pyc b/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/pointnet2_utils.cpython-39.pyc deleted file mode 100644 index 05e1f361a8470839bba54c29011af573bb44893d..0000000000000000000000000000000000000000 Binary files a/evaluations/fidkid-pytorch/3d-metrics/evals/__pycache__/pointnet2_utils.cpython-39.pyc and /dev/null differ diff --git a/evaluations/fidkid-pytorch/models/__pycache__/inception.cpython-39.pyc b/evaluations/fidkid-pytorch/models/__pycache__/inception.cpython-39.pyc deleted file mode 100644 index 36e8eb769d7138fc13ad8fc5fc6d54c4c3cda49f..0000000000000000000000000000000000000000 Binary files a/evaluations/fidkid-pytorch/models/__pycache__/inception.cpython-39.pyc and /dev/null differ diff --git a/evaluations/fidkid-pytorch/models/__pycache__/lenet.cpython-39.pyc b/evaluations/fidkid-pytorch/models/__pycache__/lenet.cpython-39.pyc deleted file mode 100644 index f01d059e16357e7e6a11eb5b01b58c568ff59e61..0000000000000000000000000000000000000000 Binary files a/evaluations/fidkid-pytorch/models/__pycache__/lenet.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/__init__.cpython-310.pyc b/guided_diffusion/__pycache__/__init__.cpython-310.pyc deleted file mode 100644 index c277829f2fb256f204c418c6fd1548c08a73cdc3..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/__init__.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/__init__.cpython-39.pyc b/guided_diffusion/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index adcca73d6d56cda9d5850e785b0f0d5e8eb7fd3f..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/continuous_diffusion.cpython-39.pyc b/guided_diffusion/__pycache__/continuous_diffusion.cpython-39.pyc deleted file mode 100644 index 2d05683073ddd2030d679ea2d61f196edfdd2145..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/continuous_diffusion.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/continuous_diffusion_utils.cpython-39.pyc b/guided_diffusion/__pycache__/continuous_diffusion_utils.cpython-39.pyc deleted file mode 100644 index c4e62a7c0cd82b2cb905b11043dd16f617b25d3a..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/continuous_diffusion_utils.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/continuous_distributions.cpython-39.pyc b/guided_diffusion/__pycache__/continuous_distributions.cpython-39.pyc deleted file mode 100644 index cee45e60342d471b7f9e54493b338c02540648eb..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/continuous_distributions.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/dist_util.cpython-310.pyc b/guided_diffusion/__pycache__/dist_util.cpython-310.pyc deleted file mode 100644 index 8413bd128494d8dcaab2c2811a8ca0de23d0eeca..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/dist_util.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/dist_util.cpython-39.pyc b/guided_diffusion/__pycache__/dist_util.cpython-39.pyc deleted file mode 100644 index f6265129153101ff474466dcb7eed1a64c29c460..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/dist_util.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/fp16_util.cpython-310.pyc b/guided_diffusion/__pycache__/fp16_util.cpython-310.pyc deleted file mode 100644 index 6361b2a18bc040170a3944bb4f72aebba834d8fe..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/fp16_util.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/fp16_util.cpython-39.pyc b/guided_diffusion/__pycache__/fp16_util.cpython-39.pyc deleted file mode 100644 index a45a0d7f241d82dcf6da67c722f473c6ee4a610f..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/fp16_util.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/gaussian_diffusion.cpython-310.pyc b/guided_diffusion/__pycache__/gaussian_diffusion.cpython-310.pyc deleted file mode 100644 index 8934b9f92778765d14d7daa1a0460a08adf8950b..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/gaussian_diffusion.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/gaussian_diffusion.cpython-39.pyc b/guided_diffusion/__pycache__/gaussian_diffusion.cpython-39.pyc deleted file mode 100644 index bdfa36f85c9df1c500b5270c7096cbf50df2a79a..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/gaussian_diffusion.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/logger.cpython-310.pyc b/guided_diffusion/__pycache__/logger.cpython-310.pyc deleted file mode 100644 index c82d6742f45671226cdf0812cc5c1a55c53c84e5..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/logger.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/logger.cpython-39.pyc b/guided_diffusion/__pycache__/logger.cpython-39.pyc deleted file mode 100644 index 1d47fa1b3cafd8e09d406c15adc9d2c32cfc1ebb..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/logger.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/losses.cpython-310.pyc b/guided_diffusion/__pycache__/losses.cpython-310.pyc deleted file mode 100644 index 714b84bc300e142c511e6db16aff107dc8b299e2..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/losses.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/losses.cpython-39.pyc b/guided_diffusion/__pycache__/losses.cpython-39.pyc deleted file mode 100644 index 9b12619094ba8a741255ed4020f8057980f1b92e..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/losses.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/nn.cpython-310.pyc b/guided_diffusion/__pycache__/nn.cpython-310.pyc deleted file mode 100644 index 4789805b14d32c559dce53310cbd98a5ce3778d2..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/nn.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/nn.cpython-39.pyc b/guided_diffusion/__pycache__/nn.cpython-39.pyc deleted file mode 100644 index 632bd30869c6f2b86b3a646e970dc0fee68bbbaf..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/nn.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/resample.cpython-310.pyc b/guided_diffusion/__pycache__/resample.cpython-310.pyc deleted file mode 100644 index 697f0a89cc85bb253e2778da1416b56d99cc9a47..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/resample.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/resample.cpython-39.pyc b/guided_diffusion/__pycache__/resample.cpython-39.pyc deleted file mode 100644 index ef0003a5becb8ba823607f9921371639ef7d75a0..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/resample.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/respace.cpython-310.pyc b/guided_diffusion/__pycache__/respace.cpython-310.pyc deleted file mode 100644 index 68f08f1c026f0ac8e4e51e2a57f4577e756c94e0..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/respace.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/respace.cpython-39.pyc b/guided_diffusion/__pycache__/respace.cpython-39.pyc deleted file mode 100644 index b1405458501a3bc931f541bc7c197de4526728af..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/respace.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/script_util.cpython-310.pyc b/guided_diffusion/__pycache__/script_util.cpython-310.pyc deleted file mode 100644 index aca621d8c286cb55c2658b6fda73fb52ad660185..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/script_util.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/script_util.cpython-39.pyc b/guided_diffusion/__pycache__/script_util.cpython-39.pyc deleted file mode 100644 index b9c342c32ccfef9b5a760aa532673965cd0cdf92..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/script_util.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/train_util.cpython-39.pyc b/guided_diffusion/__pycache__/train_util.cpython-39.pyc deleted file mode 100644 index 2fef7285b554b85574b132862b810fca8d05ec34..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/train_util.cpython-39.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/unet.cpython-310.pyc b/guided_diffusion/__pycache__/unet.cpython-310.pyc deleted file mode 100644 index dda60fb4d6a2d312027b9a8b10c3275429cbf428..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/unet.cpython-310.pyc and /dev/null differ diff --git a/guided_diffusion/__pycache__/unet.cpython-39.pyc b/guided_diffusion/__pycache__/unet.cpython-39.pyc deleted file mode 100644 index e28c6be677d4e13937034d7afcadbba9781ba2a3..0000000000000000000000000000000000000000 Binary files a/guided_diffusion/__pycache__/unet.cpython-39.pyc and /dev/null differ diff --git a/ldm/__pycache__/__init__.cpython-310.pyc b/ldm/__pycache__/__init__.cpython-310.pyc deleted file mode 100644 index 9d75f487ee6385affc95473b04622988179ec302..0000000000000000000000000000000000000000 Binary files a/ldm/__pycache__/__init__.cpython-310.pyc and /dev/null differ diff --git a/ldm/__pycache__/__init__.cpython-39.pyc b/ldm/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index adbb36a78e6489e658825738d3dccdf49d188468..0000000000000000000000000000000000000000 Binary files a/ldm/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/ldm/__pycache__/util.cpython-310.pyc b/ldm/__pycache__/util.cpython-310.pyc deleted file mode 100644 index e20edbcc7eacb222314b312c6b1a6aefd9788d2f..0000000000000000000000000000000000000000 Binary files a/ldm/__pycache__/util.cpython-310.pyc and /dev/null differ diff --git a/ldm/__pycache__/util.cpython-39.pyc b/ldm/__pycache__/util.cpython-39.pyc deleted file mode 100644 index 2dc32077f1af0b840fe2b4646f9f01ad9a1cd7af..0000000000000000000000000000000000000000 Binary files a/ldm/__pycache__/util.cpython-39.pyc and /dev/null differ diff --git a/ldm/modules/__pycache__/attention.cpython-310.pyc b/ldm/modules/__pycache__/attention.cpython-310.pyc deleted file mode 100644 index d8f6885068c56d4286242d5c363034050a376894..0000000000000000000000000000000000000000 Binary files a/ldm/modules/__pycache__/attention.cpython-310.pyc and /dev/null differ diff --git a/ldm/modules/__pycache__/attention.cpython-39.pyc b/ldm/modules/__pycache__/attention.cpython-39.pyc deleted file mode 100644 index 1d8fbfe065849056add86baf3d798e73713ba57e..0000000000000000000000000000000000000000 Binary files a/ldm/modules/__pycache__/attention.cpython-39.pyc and /dev/null differ diff --git a/ldm/modules/__pycache__/x_transformer.cpython-39.pyc b/ldm/modules/__pycache__/x_transformer.cpython-39.pyc deleted file mode 100644 index 7661ad21e0239d4942c10b2d2d30d9fba27a49f1..0000000000000000000000000000000000000000 Binary files a/ldm/modules/__pycache__/x_transformer.cpython-39.pyc and /dev/null differ diff --git a/ldm/modules/diffusionmodules/__pycache__/__init__.cpython-310.pyc b/ldm/modules/diffusionmodules/__pycache__/__init__.cpython-310.pyc deleted file mode 100644 index ce8bbae2e14542fbc5bd3b9354ccf8c7e9b4487c..0000000000000000000000000000000000000000 Binary files a/ldm/modules/diffusionmodules/__pycache__/__init__.cpython-310.pyc and /dev/null differ diff --git a/ldm/modules/diffusionmodules/__pycache__/__init__.cpython-39.pyc b/ldm/modules/diffusionmodules/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 32bedfdcdbfde1b9e55aafdacd464646dc0a447a..0000000000000000000000000000000000000000 Binary files a/ldm/modules/diffusionmodules/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/ldm/modules/diffusionmodules/__pycache__/model.cpython-39.pyc b/ldm/modules/diffusionmodules/__pycache__/model.cpython-39.pyc deleted file mode 100644 index 53733aa9c17c1c1ef7d64b9bf2253ffca8dd4558..0000000000000000000000000000000000000000 Binary files a/ldm/modules/diffusionmodules/__pycache__/model.cpython-39.pyc and /dev/null differ diff --git a/ldm/modules/diffusionmodules/__pycache__/mv_unet.cpython-39.pyc b/ldm/modules/diffusionmodules/__pycache__/mv_unet.cpython-39.pyc deleted file mode 100644 index 2aadaa6ebfecdd2cb2ae3c2c19e96bc2f4cd602c..0000000000000000000000000000000000000000 Binary files a/ldm/modules/diffusionmodules/__pycache__/mv_unet.cpython-39.pyc and /dev/null differ diff --git a/ldm/modules/diffusionmodules/__pycache__/util.cpython-310.pyc b/ldm/modules/diffusionmodules/__pycache__/util.cpython-310.pyc deleted file mode 100644 index d774a7a7606d7e505d90c701430ac44987714edc..0000000000000000000000000000000000000000 Binary files a/ldm/modules/diffusionmodules/__pycache__/util.cpython-310.pyc and /dev/null differ diff --git a/ldm/modules/diffusionmodules/__pycache__/util.cpython-39.pyc b/ldm/modules/diffusionmodules/__pycache__/util.cpython-39.pyc deleted file mode 100644 index fb5d8f1751ae97d12c8410250977fe95f841abdd..0000000000000000000000000000000000000000 Binary files a/ldm/modules/diffusionmodules/__pycache__/util.cpython-39.pyc and /dev/null differ diff --git a/ldm/modules/encoders/__pycache__/__init__.cpython-39.pyc b/ldm/modules/encoders/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index b2928472682cc5b4098ccae5ce5dca15a7ee9bf6..0000000000000000000000000000000000000000 Binary files a/ldm/modules/encoders/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/ldm/modules/encoders/__pycache__/modules.cpython-39.pyc b/ldm/modules/encoders/__pycache__/modules.cpython-39.pyc deleted file mode 100644 index ff26c58f9c4550405a2ebbe4f6e9e33c5ca3800a..0000000000000000000000000000000000000000 Binary files a/ldm/modules/encoders/__pycache__/modules.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/__init__.cpython-310.pyc b/nsr/__pycache__/__init__.cpython-310.pyc deleted file mode 100644 index e1b1045d71d20fa96e6dd96da288dd3d2fefeffd..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/__init__.cpython-310.pyc and /dev/null differ diff --git a/nsr/__pycache__/__init__.cpython-39.pyc b/nsr/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 6576ddb608daeddf2ab8b9f11ec32b5f95425809..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/camera_utils.cpython-39.pyc b/nsr/__pycache__/camera_utils.cpython-39.pyc deleted file mode 100644 index 3062b57a3d832cc696a7ae4f7fe697ecc9de165b..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/camera_utils.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/common_blks.cpython-39.pyc b/nsr/__pycache__/common_blks.cpython-39.pyc deleted file mode 100644 index c0a6c59e0fbb961ebb20a41385110131fd2b2632..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/common_blks.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/confnet.cpython-39.pyc b/nsr/__pycache__/confnet.cpython-39.pyc deleted file mode 100644 index 9988f69dafb469e84b09f884276289d0646c3100..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/confnet.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/dual_discriminator.cpython-39.pyc b/nsr/__pycache__/dual_discriminator.cpython-39.pyc deleted file mode 100644 index 2695ca73648724b80f75d3945a06c710d686a710..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/dual_discriminator.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/gs.cpython-310.pyc b/nsr/__pycache__/gs.cpython-310.pyc deleted file mode 100644 index 45b51c54b49b39a2e47ad6c2290785b8fd36e73a..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/gs.cpython-310.pyc and /dev/null differ diff --git a/nsr/__pycache__/gs.cpython-39.pyc b/nsr/__pycache__/gs.cpython-39.pyc deleted file mode 100644 index a2acc4d004e8c3e8e56c24b1d9000c26310bf90d..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/gs.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/gs_surfel.cpython-39.pyc b/nsr/__pycache__/gs_surfel.cpython-39.pyc deleted file mode 100644 index 6f5cfa65a73c662dbd89d73cf7665a9bcfcd4679..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/gs_surfel.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/networks_stylegan2.cpython-39.pyc b/nsr/__pycache__/networks_stylegan2.cpython-39.pyc deleted file mode 100644 index fc9f3e0b3720ac35b3b5310c1d57b56b864129b1..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/networks_stylegan2.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/script_util.cpython-39.pyc b/nsr/__pycache__/script_util.cpython-39.pyc deleted file mode 100644 index 2d97e4853378a34a007a21c92f9f3f062b053b42..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/script_util.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/superresolution.cpython-39.pyc b/nsr/__pycache__/superresolution.cpython-39.pyc deleted file mode 100644 index 619b316e35aa81586f83a1293ea7f010ad7c35d8..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/superresolution.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_nv_util.cpython-39.pyc b/nsr/__pycache__/train_nv_util.cpython-39.pyc deleted file mode 100644 index e89121d02b90436f1afb3b0d9635992d73b2dd7b..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_nv_util.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util.cpython-310.pyc b/nsr/__pycache__/train_util.cpython-310.pyc deleted file mode 100644 index 3e4d3fbebdc586534a41cbc25493e07078f02557..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util.cpython-310.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util.cpython-39.pyc b/nsr/__pycache__/train_util.cpython-39.pyc deleted file mode 100644 index ecf5c236d4fecefd6da45ab47d6917c39e5a3785..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_cvD.cpython-39.pyc b/nsr/__pycache__/train_util_cvD.cpython-39.pyc deleted file mode 100644 index d39b3e139e30325d5afbdbdaf8980bb3b377560b..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_cvD.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_diffusion.cpython-39.pyc b/nsr/__pycache__/train_util_diffusion.cpython-39.pyc deleted file mode 100644 index 4d5992de37201c45f5195278072e6584aefbb46c..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_diffusion.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_diffusion_dit.cpython-39.pyc b/nsr/__pycache__/train_util_diffusion_dit.cpython-39.pyc deleted file mode 100644 index 6a1bef1737162d7ab7c192171ffda908ef1792b0..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_diffusion_dit.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_diffusion_single_stage.cpython-39.pyc b/nsr/__pycache__/train_util_diffusion_single_stage.cpython-39.pyc deleted file mode 100644 index dad86cdc2ccaf5275f5a0a37086dc5f9a35740e9..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_diffusion_single_stage.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_diffusion_single_stage_sds.cpython-39.pyc b/nsr/__pycache__/train_util_diffusion_single_stage_sds.cpython-39.pyc deleted file mode 100644 index b895c8001602c0863e568ad6c5e015e4845bcdc7..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_diffusion_single_stage_sds.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_with_eg3d.cpython-39.pyc b/nsr/__pycache__/train_util_with_eg3d.cpython-39.pyc deleted file mode 100644 index 681e399518fe4438357ac4a62288e0161012526a..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_with_eg3d.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_with_eg3d_hybrid.cpython-39.pyc b/nsr/__pycache__/train_util_with_eg3d_hybrid.cpython-39.pyc deleted file mode 100644 index efb7c7ef67c2c9479388f0b0a8e2fe29f4c65772..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_with_eg3d_hybrid.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_with_eg3d_hybrid_eg3dD.cpython-39.pyc b/nsr/__pycache__/train_util_with_eg3d_hybrid_eg3dD.cpython-39.pyc deleted file mode 100644 index 7ff24dcf786694ed521367d3f386cd17d4cbd6bf..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_with_eg3d_hybrid_eg3dD.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_with_eg3d_real.cpython-39.pyc b/nsr/__pycache__/train_util_with_eg3d_real.cpython-39.pyc deleted file mode 100644 index ebdfe71dabe280d4128cb1570078a1af6fa49645..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_with_eg3d_real.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/train_util_with_eg3d_real_D.cpython-39.pyc b/nsr/__pycache__/train_util_with_eg3d_real_D.cpython-39.pyc deleted file mode 100644 index c859cdc1ec2ecf255caba0ae3dc2b5d27c98281a..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/train_util_with_eg3d_real_D.cpython-39.pyc and /dev/null differ diff --git a/nsr/__pycache__/triplane.cpython-39.pyc b/nsr/__pycache__/triplane.cpython-39.pyc deleted file mode 100644 index 6574df2e1b05945889a15b367d5dd60130af6b18..0000000000000000000000000000000000000000 Binary files a/nsr/__pycache__/triplane.cpython-39.pyc and /dev/null differ diff --git a/nsr/cvD/__pycache__/__init__.cpython-39.pyc b/nsr/cvD/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 477fb79554cf5e04e889ce34ecf0b75db0eac5a9..0000000000000000000000000000000000000000 Binary files a/nsr/cvD/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/nsr/cvD/__pycache__/canoD.cpython-39.pyc b/nsr/cvD/__pycache__/canoD.cpython-39.pyc deleted file mode 100644 index d5261c4d48b2de8e1fe1787e775c9982611dc49d..0000000000000000000000000000000000000000 Binary files a/nsr/cvD/__pycache__/canoD.cpython-39.pyc and /dev/null differ diff --git a/nsr/cvD/__pycache__/nvsD.cpython-39.pyc b/nsr/cvD/__pycache__/nvsD.cpython-39.pyc deleted file mode 100644 index efac1e81c028c7032910ca0ab1a688257f33d02a..0000000000000000000000000000000000000000 Binary files a/nsr/cvD/__pycache__/nvsD.cpython-39.pyc and /dev/null differ diff --git a/nsr/cvD/__pycache__/nvsD_canoD.cpython-39.pyc b/nsr/cvD/__pycache__/nvsD_canoD.cpython-39.pyc deleted file mode 100644 index e124a6704398b91fe0b7d93fa7cd03d0bafb7f81..0000000000000000000000000000000000000000 Binary files a/nsr/cvD/__pycache__/nvsD_canoD.cpython-39.pyc and /dev/null differ diff --git a/nsr/cvD/__pycache__/nvsD_canoD_mask.cpython-39.pyc b/nsr/cvD/__pycache__/nvsD_canoD_mask.cpython-39.pyc deleted file mode 100644 index 8515d2980d77d0158bea154af77b930ccbf703da..0000000000000000000000000000000000000000 Binary files a/nsr/cvD/__pycache__/nvsD_canoD_mask.cpython-39.pyc and /dev/null differ diff --git a/nsr/cvD/__pycache__/nvsD_canoD_multiview.cpython-39.pyc b/nsr/cvD/__pycache__/nvsD_canoD_multiview.cpython-39.pyc deleted file mode 100644 index 6b729a8d3d71e8504ee3a380d05578a0e2ece02a..0000000000000000000000000000000000000000 Binary files a/nsr/cvD/__pycache__/nvsD_canoD_multiview.cpython-39.pyc and /dev/null differ diff --git a/nsr/cvD/__pycache__/nvsD_nosr.cpython-39.pyc b/nsr/cvD/__pycache__/nvsD_nosr.cpython-39.pyc deleted file mode 100644 index d98cdaa2dbafec7a60098d4b23472d2acc471f72..0000000000000000000000000000000000000000 Binary files a/nsr/cvD/__pycache__/nvsD_nosr.cpython-39.pyc and /dev/null differ diff --git a/nsr/losses/__pycache__/__init__.cpython-39.pyc b/nsr/losses/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 3f11a0151d6fdd02b14c820aaacbd483ab85fa14..0000000000000000000000000000000000000000 Binary files a/nsr/losses/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/nsr/losses/__pycache__/builder.cpython-39.pyc b/nsr/losses/__pycache__/builder.cpython-39.pyc deleted file mode 100644 index a5e7adb9cc7097c6a932991d617a7b8ed47f729b..0000000000000000000000000000000000000000 Binary files a/nsr/losses/__pycache__/builder.cpython-39.pyc and /dev/null differ diff --git a/nsr/losses/__pycache__/helpers.cpython-39.pyc b/nsr/losses/__pycache__/helpers.cpython-39.pyc deleted file mode 100644 index 32ae81bd8e067fc571c072a97f44d4ca8914f394..0000000000000000000000000000000000000000 Binary files a/nsr/losses/__pycache__/helpers.cpython-39.pyc and /dev/null differ diff --git a/nsr/losses/__pycache__/id_loss.cpython-39.pyc b/nsr/losses/__pycache__/id_loss.cpython-39.pyc deleted file mode 100644 index a74c2872bf4df5dffdc148190a8a7d36a6d78886..0000000000000000000000000000000000000000 Binary files a/nsr/losses/__pycache__/id_loss.cpython-39.pyc and /dev/null differ diff --git a/nsr/losses/__pycache__/model_irse.cpython-39.pyc b/nsr/losses/__pycache__/model_irse.cpython-39.pyc deleted file mode 100644 index 4e36976164241d573cb616d29d890b96423d540e..0000000000000000000000000000000000000000 Binary files a/nsr/losses/__pycache__/model_irse.cpython-39.pyc and /dev/null differ diff --git a/nsr/losses/__pycache__/paths_config.cpython-39.pyc b/nsr/losses/__pycache__/paths_config.cpython-39.pyc deleted file mode 100644 index 6af4b03c1c41e8146c83f09c6d0305d9d09777d7..0000000000000000000000000000000000000000 Binary files a/nsr/losses/__pycache__/paths_config.cpython-39.pyc and /dev/null differ diff --git a/nsr/losses/__pycache__/sdfstudio_losses.cpython-39.pyc b/nsr/losses/__pycache__/sdfstudio_losses.cpython-39.pyc deleted file mode 100644 index adb9f32b9f214892ac015898da401cd8c8188b6e..0000000000000000000000000000000000000000 Binary files a/nsr/losses/__pycache__/sdfstudio_losses.cpython-39.pyc and /dev/null differ diff --git a/nsr/losses/__pycache__/vqperceptual.cpython-39.pyc b/nsr/losses/__pycache__/vqperceptual.cpython-39.pyc deleted file mode 100644 index 2f597ed846a4a313efb3b8f15209b678fba2e562..0000000000000000000000000000000000000000 Binary files a/nsr/losses/__pycache__/vqperceptual.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/__init__.cpython-39.pyc b/nsr/lsgm/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index dcc5d6633692797e7e7d186a7a45e39578e6fac9..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/controlLDM.cpython-39.pyc b/nsr/lsgm/__pycache__/controlLDM.cpython-39.pyc deleted file mode 100644 index 1c3392091760843eb6e06d2a507d88c2c487171a..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/controlLDM.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/crossattn_cldm.cpython-39.pyc b/nsr/lsgm/__pycache__/crossattn_cldm.cpython-39.pyc deleted file mode 100644 index 81c0f0bd024da709cf1ceca030ecfebed0c5754b..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/crossattn_cldm.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/flow_matching_trainer.cpython-39.pyc b/nsr/lsgm/__pycache__/flow_matching_trainer.cpython-39.pyc deleted file mode 100644 index fce3a6e67147402c1bfa0900664bea3b0942dfe2..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/flow_matching_trainer.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/sgm_DiffusionEngine.cpython-39.pyc b/nsr/lsgm/__pycache__/sgm_DiffusionEngine.cpython-39.pyc deleted file mode 100644 index 614c07f68455b46fa361ec4a94be395df113c724..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/sgm_DiffusionEngine.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/train_util_diffusion_lsgm.cpython-39.pyc b/nsr/lsgm/__pycache__/train_util_diffusion_lsgm.cpython-39.pyc deleted file mode 100644 index 215efd1a58a4239a945d29c1c9a3f789c27443b0..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/train_util_diffusion_lsgm.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/train_util_diffusion_lsgm_cvD_joint.cpython-39.pyc b/nsr/lsgm/__pycache__/train_util_diffusion_lsgm_cvD_joint.cpython-39.pyc deleted file mode 100644 index 87e2ba15ce039f5dfb00a282f2d8b94d6369a804..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/train_util_diffusion_lsgm_cvD_joint.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/train_util_diffusion_lsgm_noD.cpython-39.pyc b/nsr/lsgm/__pycache__/train_util_diffusion_lsgm_noD.cpython-39.pyc deleted file mode 100644 index 67e27533edb0f3c95e499292268b3ce55f575e52..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/train_util_diffusion_lsgm_noD.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/train_util_diffusion_lsgm_noD_joint.cpython-39.pyc b/nsr/lsgm/__pycache__/train_util_diffusion_lsgm_noD_joint.cpython-39.pyc deleted file mode 100644 index 9a0525ae93a1f84ad8c3859bfa9bc49f0a21ad9a..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/train_util_diffusion_lsgm_noD_joint.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/__pycache__/train_util_diffusion_vpsde.cpython-39.pyc b/nsr/lsgm/__pycache__/train_util_diffusion_vpsde.cpython-39.pyc deleted file mode 100644 index fc69a4f7243a5d0807841ae3e83eeb6acd6b438f..0000000000000000000000000000000000000000 Binary files a/nsr/lsgm/__pycache__/train_util_diffusion_vpsde.cpython-39.pyc and /dev/null differ diff --git a/nsr/lsgm/flow_matching_trainer.py b/nsr/lsgm/flow_matching_trainer.py index 91acd9f2e30d86debce0a39dff627efb83171f8c..85ef153369ea7e306d5e8ceefc822d24de13dff5 100644 --- a/nsr/lsgm/flow_matching_trainer.py +++ b/nsr/lsgm/flow_matching_trainer.py @@ -23,6 +23,7 @@ import numpy as np import torch as th import torch.distributed as dist import torchvision +from nsr.camera_utils import generate_input_camera, uni_mesh_path from PIL import Image from torch.nn.parallel.distributed import DistributedDataParallel as DDP from torch.optim import AdamW @@ -45,7 +46,6 @@ from guided_diffusion.train_util import (TrainLoop, calc_average_loss, from guided_diffusion.gaussian_diffusion import ModelMeanType from ldm.modules.encoders.modules import FrozenClipImageEmbedder, TextEmbedder, FrozenCLIPTextEmbedder, FrozenOpenCLIPImagePredictionEmbedder, FrozenOpenCLIPImageEmbedder -from nsr.camera_utils import generate_input_camera, uni_mesh_path import dnnlib from dnnlib.util import requires_grad @@ -63,6 +63,51 @@ from .train_util_diffusion_lsgm_noD_joint import TrainLoop3DDiffusionLSGMJointno from .crossattn_cldm import TrainLoop3DDiffusionLSGM_crossattn +# Function to generate a rotation matrix for an arbitrary theta along the x-axis +def rotation_matrix_x(theta_degrees): + theta = np.radians(theta_degrees) # Convert degrees to radians + cos_theta = np.cos(theta) + sin_theta = np.sin(theta) + + rotation_matrix = np.array([[1, 0, 0], + [0, cos_theta, -sin_theta], + [0, sin_theta, cos_theta]]) + return rotation_matrix + +def rotation_matrix_z(theta): + """ + Returns a 3x3 rotation matrix that rotates a point around the z-axis by theta radians. + + Parameters: + theta (float): The angle of rotation in radians. + + Returns: + numpy.ndarray: A 3x3 rotation matrix. + """ + return np.array([ + [np.cos(theta), -np.sin(theta), 0], + [np.sin(theta), np.cos(theta), 0], + [0, 0, 1] + ]) + +def rotation_matrix_y(theta): + """ + Returns a 3x3 rotation matrix that rotates a point around the y-axis by theta radians. + + Parameters: + theta (float): The angle of rotation in radians. + + Returns: + numpy.ndarray: A 3x3 rotation matrix. + """ + return np.array([ + [np.cos(theta), 0, np.sin(theta)], + [0, 1, 0 ], + [-np.sin(theta), 0, np.cos(theta)] + ]) + + + # import SD stuffs from typing import Any, Dict, List, Optional, Tuple, Union from contextlib import contextmanager @@ -85,18 +130,6 @@ import open3d as o3d # from sgm.sampling_utils.demo.streamlit_helpers import init_sampling -# Function to generate a rotation matrix for an arbitrary theta along the x-axis -def rotation_matrix_x(theta_degrees): - theta = np.radians(theta_degrees) # Convert degrees to radians - cos_theta = np.cos(theta) - sin_theta = np.sin(theta) - - rotation_matrix = np.array([[1, 0, 0], - [0, cos_theta, -sin_theta], - [0, sin_theta, cos_theta]]) - return rotation_matrix - - def sample_uniform_cameras_on_sphere(num_samples=1): # Step 1: Sample azimuth angles uniformly from [0, 2*pi) theta = np.random.uniform(0, 2 * np.pi, num_samples) @@ -678,12 +711,11 @@ class FlowMatchingEngine(TrainLoop3DDiffusionLSGM_crossattn): seed=42, **kwargs, ): - - logger.log(f'cfg_scale: {cfg_scale}, seed: {seed}') - # self.sampler sample_fn = self.transport_sampler.sample_ode(num_steps=250, cfg=True) # default ode sampling setting. + logger.log(f'cfg_scale: {cfg_scale}, seed: {seed}') + th.manual_seed(seed) # to reproduce result zs = th.randn(batch_size, *shape).to(dist_util.dev()).to(self.dtype) # st() @@ -990,14 +1022,14 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): ): self.ddpm_model.eval() - # args = dnnlib.EasyDict( - # dict( - # batch_size=1, - # image_size=self.diffusion_input_size, - # denoise_in_channels=self.rec_model.decoder.triplane_decoder. - # out_chans, # type: ignore - # clip_denoised=False, - # class_cond=False)) + args = dnnlib.EasyDict( + dict( + batch_size=1, + image_size=self.diffusion_input_size, + denoise_in_channels=self.rec_model.decoder.triplane_decoder. + out_chans, # type: ignore + clip_denoised=False, + class_cond=False)) model_kwargs = {} @@ -1248,7 +1280,7 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): # name = f'{idx}-{i}-fuse.ply' # name = f'mesh.obj' - name = f'{idx}/mesh_raw.obj' + name = f'{idx}/{i}-mesh_raw.obj' # st() # depth_trunc = (radius * 2.0) if depth_trunc < 0 else depth_trunc # voxel_size = (depth_trunc / mesh_res) if voxel_size < 0 else voxel_size @@ -1260,17 +1292,19 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): logger.log("mesh saved at {}".format(os.path.join(train_dir, name))) # post-process the mesh and save, saving the largest N clusters # mesh_post = post_process_mesh(mesh, cluster_to_keep=num_cluster) - mesh_post = post_process_mesh(mesh) - # o3d.io.write_triangle_mesh(os.path.join(train_dir, name.replace('.obj', '_post.obj')), mesh_post) + mesh_post = post_process_mesh(mesh) mesh_vertices = np.asarray(mesh_post.vertices) # Convert vertices to a numpy array rotated_vertices = mesh_vertices @ rotation_matrix_x(-90).T + # rotated_vertices = rotated_vertices @ rotation_matrix_z(np.pi).T + rotated_vertices = rotated_vertices @ rotation_matrix_y(np.pi).T mesh_post.vertices = o3d.utility.Vector3dVector(rotated_vertices) # Update vertices post_mesh_path = os.path.join(train_dir, name.replace('_raw.obj', '.obj')) o3d.io.write_triangle_mesh(post_mesh_path, mesh_post) logger.log("mesh post processed saved at {}".format(post_mesh_path)) + return post_mesh_path @@ -1287,16 +1321,19 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): return o3d.mesh """ - if self.aabb is not None: # as in lara. - center = self.aabb.mean(0) - # radius = np.linalg.norm(self.aabb[1] - self.aabb[0]) * 0.5 - radius = np.linalg.norm(self.aabb[1] - self.aabb[0]) * 0.5 - # voxel_size = radius / 256 - # voxel_size = radius / 192 # less holes - voxel_size = radius / 160 # smaller - # sdf_trunc = voxel_size * 16 # less holes, slower integration - sdf_trunc = voxel_size * 12 # - print("using aabb") + # if self.aabb is not None: # as in lara. + # center = self.aabb.mean(0) + # radius = np.linalg.norm(self.aabb[1] - self.aabb[0]) * 0.5 + # voxel_size = radius / 256 + # sdf_trunc = voxel_size * 2 + # print("using aabb") + + assert self.aabb is not None # as in lara. + center = self.aabb.mean(0) + radius = np.linalg.norm(self.aabb[1] - self.aabb[0]) * 0.5 + voxel_size = radius / 160 # less holes + sdf_trunc = voxel_size * 12 + print("using aabb") volume = o3d.pipelines.integration.ScalableTSDFVolume( voxel_length= voxel_size, @@ -1362,6 +1399,9 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): output_dir=None, for_fid=False,): + if output_dir is None: + output_dir = logger.get_dir() + batch_size, L, C = planes.shape # ddpm_latent = { self.latent_name: planes[..., :-3] * self.feat_scale_factor.to(planes), # kl-reg latent @@ -1401,8 +1441,20 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): # pcu.save_mesh_v(f'{output_dir}/gaussian.ply', ddpm_latent['gaussians_upsampled'][0, ..., :3].cpu().numpy()) # fps-downsampling? - pred_gaussians_xyz = ddpm_latent['gaussians_upsampled_3'][..., :3] + # pred_gaussians_xyz = ddpm_latent['gaussians_upsampled_3'][..., :3] fine_gs = ddpm_latent[fine_scale] + + fine_gs_numpy = fine_gs.cpu().numpy() + vtx = np.transpose(rotation_matrix_x(-90) @ np.transpose(fine_gs_numpy[0, :, :3])) # for gradio visualization + # vtx = vtx @ rotation_matrix_z(np.pi).T + vtx = vtx @ rotation_matrix_y(np.pi).T + cloud = trimesh.PointCloud(vtx, colors=fine_gs_numpy[0, :, 10:13]) + # Save the point cloud to an OBJ file + rgb_xyz_path_forgradio = f'{output_dir}/{name_prefix}-gaussian-pcd.glb' # gradio only accepts glb for visualization + _ = cloud.export(rgb_xyz_path_forgradio) + + rgb_xyz_path_formeshlab = f'{output_dir}/{name_prefix}-gaussian-pcd.ply' # for meshlab visualization + _ = cloud.export(rgb_xyz_path_formeshlab) # K=4096 # query_pcd_xyz, fps_idx = pytorch3d.ops.sample_farthest_points( @@ -1410,24 +1462,13 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): # # random_start_point=False) # B self.latent_num # random_start_point=True) # B self.latent_num - if output_dir is None: - output_dir = logger.get_dir() # pcu.save_mesh_v(f'{output_dir}/{name_prefix}-gaussian-{K}.ply', query_pcd_xyz[0].cpu().numpy()) - np.save(f'{output_dir}/{name_prefix}-gaussian.npy', fine_gs.cpu().numpy()[0]) # L, 13 - - # also save rgb point cloud - fine_gs_numpy = fine_gs.cpu().numpy() - rgb_xyz_path = f'{output_dir}/{name_prefix}-gaussian-pcd.glb' - # pcu.save_mesh_vc(rgb_xyz_path, v=fine_gs_numpy[..., :3], c=fine_gs_numpy[..., 3:6]) - vtx = np.transpose(rotation_matrix_x(-90) @ np.transpose(fine_gs_numpy[0, :, :3])) - cloud = trimesh.PointCloud(vtx, colors=fine_gs_numpy[0, :, 10:13]) - # Save the point cloud to an OBJ file - _ = cloud.export(rgb_xyz_path) + np.save(f'{output_dir}/{name_prefix}-gaussian.npy', fine_gs_numpy) + video_path = f'{output_dir}/{name_prefix}-gs.mp4' # return None, None - video_path = f'{output_dir}/{name_prefix}-gs.mp4' try: # video_out = imageio.get_writer( @@ -1476,8 +1517,8 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): ''' assert render_reference is not None - # render_reference = th.load('eval_pose.pt', map_location='cpu').numpy()[:24] + # render_reference = th.load('eval_pose.pt', map_location='cpu').numpy()[:24] # rand_start_idx = random.randint(0,2) # render_reference = render_reference[rand_start_idx::3] # randomly render 8 views, maintain fixed azimuths # assert len(render_reference)==8 @@ -1518,6 +1559,23 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): # ! if visualizing a single scale fine_scale_key = list(pred.keys())[-1] + # pred = pred[fine_scale_key] + + # for k in pred.keys(): + # pred[k] = einops.rearrange(pred[k], 'B V ... -> (B V) ...') # merge + + # pred_vis = self._make_vis_img(pred) + + # vis = pred_vis.permute(0, 2, 3, 1).cpu().numpy() + # vis = vis * 127.5 + 127.5 + # vis = vis.clip(0, 255).astype(np.uint8) + + # # if not save_img: + # for j in range(vis.shape[0] + # ): # ! currently only export one plane at a time + # video_out.append_data(vis[j]) + + # save multi-scale rendering all_rgbs.append(einops.rearrange(pred[fine_scale_key]['image'], 'B V ... -> (B V) ...')) all_depths.append(einops.rearrange(pred[fine_scale_key]['depth'], 'B V ... -> (B V) ...')) @@ -1525,13 +1583,12 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): all_pred_vis = {} # for key in pred.keys(): - # st() for key in ['gaussians_base', fine_scale_key]: # only show two LoDs pred_scale = pred[key] # only show finest result here for k in pred_scale.keys(): pred_scale[k] = einops.rearrange(pred_scale[k], 'B V ... -> (B V) ...') # merge - pred_vis = self._make_vis_img(pred_scale) + pred_vis = self._make_vis_img(pred_scale, ignore_depth=True) vis = pred_vis.permute(0, 2, 3, 1).cpu().numpy() vis = vis * 127.5 + 127.5 @@ -1539,7 +1596,13 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): all_pred_vis[key] = vis - all_pred_vis_concat = np.concatenate([cv2.resize(all_pred_vis[k][0], (512*3, 512)) for k in all_pred_vis.keys()], axis=0) + # all_pred_vis_concat = np.concatenate([cv2.resize(all_pred_vis[k][0], (384*3, 384)) for k in ['gaussians_base', 'gaussians_upsampled', 'gaussians_upsampled_2']], axis=0) + # all_pred_vis_concat = np.concatenate([cv2.resize(all_pred_vis[k][0], (256*3, 256)) for k in ['gaussians_base', 'gaussians_upsampled', 'gaussians_upsampled_2']], axis=0) + # all_pred_vis_concat = np.concatenate([cv2.resize(all_pred_vis[k][0], (384*len(all_pred_vis.keys()), 384)) for k in all_pred_vis.keys()], axis=0) + # all_pred_vis_concat = np.concatenate([cv2.resize(all_pred_vis[k][0], (384*3, 384)) for k in all_pred_vis.keys()], axis=0) + + # all_pred_vis_concat = np.concatenate([cv2.resize(all_pred_vis[k][0], (512*3, 512)) for k in all_pred_vis.keys()], axis=0) + all_pred_vis_concat = np.concatenate([cv2.resize(all_pred_vis[k][0], (512*2, 512)) for k in all_pred_vis.keys()], axis=0) video_out.append_data(all_pred_vis_concat) @@ -1557,31 +1620,40 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): f'{output_dir}/{name_prefix}.mp4') del video_out, pred, pred_vis, vis - return all_rgbs, all_depths, all_alphas, video_path, rgb_xyz_path + # return all_rgbs, all_depths, all_alphas + return all_rgbs, all_depths, all_alphas, video_path, rgb_xyz_path_forgradio @th.no_grad() - def _make_vis_img(self, pred): + def _make_vis_img(self, pred, ignore_depth=False): + gen_img = pred['image_raw'] + rend_normal = pred['rend_normal'] # if True: - pred_depth = pred['image_depth'] - pred_depth = (pred_depth - pred_depth.min()) / (pred_depth.max() - - pred_depth.min()) + if not ignore_depth: + pred_depth = pred['image_depth'] + pred_depth = (pred_depth - pred_depth.min()) / (pred_depth.max() - + pred_depth.min()) - pred_depth = pred_depth.cpu()[0].permute(1, 2, 0).numpy() - pred_depth = (plt.cm.viridis(pred_depth[..., 0])[..., :3]) * 2 - 1 - pred_depth = th.from_numpy(pred_depth).to( - pred['image_raw'].device).permute(2, 0, 1).unsqueeze(0) + pred_depth = pred_depth.cpu()[0].permute(1, 2, 0).numpy() + pred_depth = (plt.cm.viridis(pred_depth[..., 0])[..., :3]) * 2 - 1 + pred_depth = th.from_numpy(pred_depth).to( + pred['image_raw'].device).permute(2, 0, 1).unsqueeze(0) - gen_img = pred['image_raw'] - rend_normal = pred['rend_normal'] + pred_vis = th.cat( + [ + gen_img, + rend_normal, + pred_depth, + ], + dim=-1) # B, 3, H, W - pred_vis = th.cat( - [ - gen_img, - rend_normal, - pred_depth, - ], - dim=-1) # B, 3, H, W + else: + pred_vis = th.cat( + [ + gen_img, + rend_normal, + ], + dim=-1) # B, 3, H, W return pred_vis @@ -1658,18 +1730,16 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): # # ! todo, transform to gs camera if self.latent_key != 'latent': # normalized-xyz - + pcd_export_dir = f'{save_dir}/{name_prefix}.glb' # pcu fails on py=3.9 vtx = self.unnormalize_pcd_act(samples[i]).detach().cpu().float().numpy() - cloud = trimesh.PointCloud(vtx @ rotation_matrix_x(-90).T, colors=np.ones_like(vtx)*0.75) + cloud = trimesh.PointCloud(vtx @ rotation_matrix_x(-90).T, colors=np.ones_like(vtx)*0.1) # since white background _ = cloud.export(pcd_export_dir) # for gradio display logger.log(f'stage-1 glb point cloud saved to {pcd_export_dir}') - pcd_export_dir_forstage1 = f'{save_dir}/{name_prefix}.ply' # pcu fails on py=3.9 - cloud = trimesh.PointCloud(vtx) - _ = cloud.export(pcd_export_dir_forstage1) # for stage-1 cascaded loading. - - logger.log(f'point cloud saved to {pcd_export_dir_forstage1}') + pcd_export_dir_forstage1 = f'{save_dir}/{name_prefix}.ply' + pcu.save_mesh_v(pcd_export_dir_forstage1, self.unnormalize_pcd_act(samples[i]).detach().cpu().float().numpy()) + logger.log(f'point cloud saved to {pcd_export_dir}') return pcd_export_dir else: # ! editing debug @@ -1782,12 +1852,11 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): # # ! todo, transform to gs camera if self.latent_key != 'latent': # normalized-xyz - stage1_pcd_path = f'{logger.get_dir()}/{name_prefix}.glb' - pcu.save_mesh_v( stage1_pcd_path, self.unnormalize_pcd_act(samples[i]).detach().cpu().float().numpy()) - logger.log(f'point cloud saved to {stage1_pcd_path}') + pcu.save_mesh_v( f'{logger.get_dir()}/{name_prefix}.ply', self.unnormalize_pcd_act(samples[i]).detach().cpu().float().numpy()) + logger.log(f'point cloud saved to {logger.get_dir()}/{name_prefix}.ply') else: # ! editing debug - all_rgbs, all_depths, all_alphas, _, _ = self.render_gs_video_given_latent( + all_rgbs, all_depths, all_alphas = self.render_gs_video_given_latent( # samples[i:i+1].to(self.dtype), # default version # th.cat([gt_kl_latent.to(samples), gt_xyz.to(samples)], dim=-1), @@ -1804,8 +1873,8 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): render_reference=batch, export_mesh=False) - # if export_mesh: - # self.export_mesh_from_2dgs(all_rgbs, all_depths, idx, i) + if export_mesh: + self.export_mesh_from_2dgs(all_rgbs, all_depths, idx, i) if self.cond_key == 'caption': assert prompt != '' @@ -1875,30 +1944,120 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): self.sample_and_save(batch_c, ucg_keys, num_samples, camera,) + @th.inference_mode() + def eval_i23d_and_export_gradio( + self, + inp_img, + seed=42, + cfg_scale=4.0, # default value in neural ode + save_img=False, + **kwargs, + ): + + # logger.log( + # unconditional_guidance_scale, + # seed, + # mesh_size, + # mesh_thres, + # ) + + sampling_kwargs = { + 'cfg_scale': cfg_scale, # default value in SiT + 'seed': seed, + } + + camera = th.load('assets/objv_eval_pose.pt', map_location=dist_util.dev())[:24] + inp_img = th.from_numpy(inp_img).permute(2,0,1).unsqueeze(0) / 127.5 - 1 # to [-1,1] + + num_samples=1 + export_mesh=True + + # self.ddpm_model.eval() + + # args = dnnlib.EasyDict( + # dict( + # batch_size=1, + # image_size=self.diffusion_input_size, + # denoise_in_channels=self.rec_model.decoder.triplane_decoder. + # out_chans, # type: ignore + # clip_denoised=False, + # class_cond=False)) + + # model_kwargs = {} + + ucg_keys = [self.cond_key] # i23d + + ins_name = 'house2-input' # for debug here + + if self.cond_key == 'img-xyz': # stage-2 + + i = 0 # for gradio only + # for i in range(1): + stage_1_output_dir="./logs/i23d/stage-1/dino_img/" + stage1_pcd_output_path = f'{stage_1_output_dir}/{ins_name}/sample-0-{i}.ply' + + fps_xyz = trimesh.load(stage1_pcd_output_path).vertices # pcu may fail on py=3.9 + fps_xyz = torch.from_numpy(fps_xyz).clip(-0.45,0.45).unsqueeze(0) + + logger.log('loading stage-1 point cloud from: ', stage1_pcd_output_path) + + # fps_xyz = None # ! TODO, load from local directory + # batch_c = { + # 'img': batch['img'][0:1].to(self.dtype).to(dist_util.dev()), + # 'fps-xyz': fps_xyz[0:1].to(self.dtype).to(dist_util.dev()), + # } + batch_c = {'img': inp_img.to(dist_util.dev()).to(self.dtype), + 'fps-xyz': fps_xyz[0:1].to(self.dtype).to(dist_util.dev())} + + # no need to return here? + video_path, rgb_xyz_path, post_mesh_path = self.sample_and_save(batch_c, ucg_keys, num_samples, camera, save_img, idx=ins_name, export_mesh=export_mesh, stage1_idx=i, **sampling_kwargs) # type: ignore + + # video_path = './logs/i23d/stage-2/dino_img/house2-input/sample-0-0-gs.mp4' + # rgb_xyz_path = './logs/i23d/stage-2/dino_img/low-poly-model-of-a-green-pine-tree,-also-resembling-a-Christmas-tree.-vc.ply' + assert post_mesh_path != '' + + return video_path, rgb_xyz_path, post_mesh_path + + else: # stage-1 data + # batch_c = {self.cond_key: batch[self.cond_key][0:1].to(dist_util.dev()).to(self.dtype), } + + # raise NotImplementedError('stage-2 only') + batch_c = {'img': inp_img.to(dist_util.dev()).to(self.dtype)} + pcd_export_dir = self.sample_and_save(batch_c, ucg_keys, num_samples, camera, save_img, idx=ins_name, export_mesh=export_mesh, **sampling_kwargs) # type: ignore + return pcd_export_dir + + @th.inference_mode() def eval_i23d_and_export( self, prompt="Yellow rubber duck", + # use_ddim=False, + unconditional_guidance_scale=4.0, save_img=False, - use_train_trajectory=False, + seed=42, + # cfg_scale=4.0, # default value in neural ode camera=None, num_samples=1, stage_1_output_dir='', - num_instances=1, + # num_instances=1, export_mesh=False, ): self.ddpm_model.eval() - args = dnnlib.EasyDict( - dict( - batch_size=1, - image_size=self.diffusion_input_size, - denoise_in_channels=self.rec_model.decoder.triplane_decoder. - out_chans, # type: ignore - clip_denoised=False, - class_cond=False)) + # args = dnnlib.EasyDict( + # dict( + # batch_size=1, + # image_size=self.diffusion_input_size, + # denoise_in_channels=self.rec_model.decoder.triplane_decoder. + # out_chans, # type: ignore + # clip_denoised=False, + # class_cond=False)) - model_kwargs = {} + # model_kwargs = {} + sampling_kwargs = { + 'cfg_scale': unconditional_guidance_scale, # default value in SiT + 'seed': seed, + } uc = None log = dict() @@ -1969,7 +2128,7 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): 'fps-xyz': fps_xyz[0:1].to(self.dtype).to(dist_util.dev()), } - self.sample_and_save(batch_c, ucg_keys, num_samples, camera, save_img, idx=ins_name, export_mesh=export_mesh, stage1_idx=i) # type: ignore + self.sample_and_save(batch_c, ucg_keys, num_samples, camera, save_img, idx=ins_name, export_mesh=export_mesh, stage1_idx=i,**sampling_kwargs) # type: ignore else: # stage-1 data batch_c = {self.cond_key: batch[self.cond_key][0:1].to(dist_util.dev()).to(self.dtype), } @@ -1984,93 +2143,11 @@ class FlowMatchingEngine_gs(FlowMatchingEngine): # os.mkdir(save_dir, exists_ok=True, parents=True) # self.sample_and_save(batch_c, ucg_keys, num_samples, camera, save_img, idx=f'{idx}-{ins}', export_mesh=export_mesh) - self.sample_and_save(batch_c, ucg_keys, num_samples, camera, save_img, idx=ins_name, export_mesh=export_mesh) # type: ignore + self.sample_and_save(batch_c, ucg_keys, num_samples, camera, save_img, idx=ins_name, export_mesh=export_mesh,**sampling_kwargs) # type: ignore gc.collect() - @th.inference_mode() - def eval_i23d_and_export_gradio( - self, - inp_img, - seed=42, - cfg_scale=4.0, # default value in neural ode - save_img=False, - **kwargs, - ): - - # logger.log( - # unconditional_guidance_scale, - # seed, - # mesh_size, - # mesh_thres, - # ) - - sampling_kwargs = { - 'cfg_scale': cfg_scale, # default value in SiT - 'seed': seed, - } - - camera = th.load('assets/objv_eval_pose.pt', map_location=dist_util.dev())[:24] - inp_img = th.from_numpy(inp_img).permute(2,0,1).unsqueeze(0) / 127.5 - 1 # to [-1,1] - - num_samples=1 - export_mesh=True - - # self.ddpm_model.eval() - - # args = dnnlib.EasyDict( - # dict( - # batch_size=1, - # image_size=self.diffusion_input_size, - # denoise_in_channels=self.rec_model.decoder.triplane_decoder. - # out_chans, # type: ignore - # clip_denoised=False, - # class_cond=False)) - - # model_kwargs = {} - - ucg_keys = [self.cond_key] # i23d - - ins_name = 'house2-input' # for debug here - - if self.cond_key == 'img-xyz': # stage-2 - - i = 0 # for gradio only - # for i in range(1): - stage_1_output_dir="./logs/i23d/stage-1/dino_img/" - stage1_pcd_output_path = f'{stage_1_output_dir}/{ins_name}/sample-0-{i}.ply' - - fps_xyz = trimesh.load(stage1_pcd_output_path).vertices # pcu may fail on py=3.9 - fps_xyz = torch.from_numpy(fps_xyz).clip(-0.45,0.45).unsqueeze(0) - - logger.log('loading stage-1 point cloud from: ', stage1_pcd_output_path) - - # fps_xyz = None # ! TODO, load from local directory - # batch_c = { - # 'img': batch['img'][0:1].to(self.dtype).to(dist_util.dev()), - # 'fps-xyz': fps_xyz[0:1].to(self.dtype).to(dist_util.dev()), - # } - batch_c = {'img': inp_img.to(dist_util.dev()).to(self.dtype), - 'fps-xyz': fps_xyz[0:1].to(self.dtype).to(dist_util.dev())} - - # no need to return here? - video_path, rgb_xyz_path, post_mesh_path = self.sample_and_save(batch_c, ucg_keys, num_samples, camera, save_img, idx=ins_name, export_mesh=export_mesh, stage1_idx=i, **sampling_kwargs) # type: ignore - - # video_path = './logs/i23d/stage-2/dino_img/house2-input/sample-0-0-gs.mp4' - # rgb_xyz_path = './logs/i23d/stage-2/dino_img/low-poly-model-of-a-green-pine-tree,-also-resembling-a-Christmas-tree.-vc.ply' - assert post_mesh_path != '' - - return video_path, rgb_xyz_path, post_mesh_path - - else: # stage-1 data - # batch_c = {self.cond_key: batch[self.cond_key][0:1].to(dist_util.dev()).to(self.dtype), } - - # raise NotImplementedError('stage-2 only') - batch_c = {'img': inp_img.to(dist_util.dev()).to(self.dtype)} - pcd_export_dir = self.sample_and_save(batch_c, ucg_keys, num_samples, camera, save_img, idx=ins_name, export_mesh=export_mesh, **sampling_kwargs) # type: ignore - return pcd_export_dir - diff --git a/nsr/srt/__pycache__/__init__.cpython-39.pyc b/nsr/srt/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index c7504e5b42a6355429c60e9c0ef5f8992422e169..0000000000000000000000000000000000000000 Binary files a/nsr/srt/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/nsr/srt/__pycache__/encoder.cpython-39.pyc b/nsr/srt/__pycache__/encoder.cpython-39.pyc deleted file mode 100644 index d943f1788cdec7c2b013dd910d394ed4b166ee77..0000000000000000000000000000000000000000 Binary files a/nsr/srt/__pycache__/encoder.cpython-39.pyc and /dev/null differ diff --git a/nsr/srt/__pycache__/layers.cpython-39.pyc b/nsr/srt/__pycache__/layers.cpython-39.pyc deleted file mode 100644 index 218dd884c95bf263d84cbdab42b9ba31401bfb9c..0000000000000000000000000000000000000000 Binary files a/nsr/srt/__pycache__/layers.cpython-39.pyc and /dev/null differ diff --git a/nsr/volumetric_rendering/__pycache__/__init__.cpython-39.pyc b/nsr/volumetric_rendering/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 7d359c2530e776e39735111c1fe408f2a4bcf622..0000000000000000000000000000000000000000 Binary files a/nsr/volumetric_rendering/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/nsr/volumetric_rendering/__pycache__/math_utils.cpython-39.pyc b/nsr/volumetric_rendering/__pycache__/math_utils.cpython-39.pyc deleted file mode 100644 index 393a45ba78b68411e9abb826561d92a27fb94cf7..0000000000000000000000000000000000000000 Binary files a/nsr/volumetric_rendering/__pycache__/math_utils.cpython-39.pyc and /dev/null differ diff --git a/nsr/volumetric_rendering/__pycache__/ray_marcher.cpython-39.pyc b/nsr/volumetric_rendering/__pycache__/ray_marcher.cpython-39.pyc deleted file mode 100644 index 287c0da0adcd9dd33c35592718d8d106b02c8bb6..0000000000000000000000000000000000000000 Binary files a/nsr/volumetric_rendering/__pycache__/ray_marcher.cpython-39.pyc and /dev/null differ diff --git a/nsr/volumetric_rendering/__pycache__/ray_sampler.cpython-39.pyc b/nsr/volumetric_rendering/__pycache__/ray_sampler.cpython-39.pyc deleted file mode 100644 index e839a50129afc2b418813afb363bf5ea0031349a..0000000000000000000000000000000000000000 Binary files a/nsr/volumetric_rendering/__pycache__/ray_sampler.cpython-39.pyc and /dev/null differ diff --git a/nsr/volumetric_rendering/__pycache__/renderer.cpython-39.pyc b/nsr/volumetric_rendering/__pycache__/renderer.cpython-39.pyc deleted file mode 100644 index bd2c7fa3f14698fb3350a5832513befb269a9569..0000000000000000000000000000000000000000 Binary files a/nsr/volumetric_rendering/__pycache__/renderer.cpython-39.pyc and /dev/null differ diff --git a/scripts/__pycache__/run_0123pp.cpython-39.pyc b/scripts/__pycache__/run_0123pp.cpython-39.pyc deleted file mode 100644 index c07eed76d0b4584dc909236b00f174323f146382..0000000000000000000000000000000000000000 Binary files a/scripts/__pycache__/run_0123pp.cpython-39.pyc and /dev/null differ diff --git a/sgm/__pycache__/__init__.cpython-39.pyc b/sgm/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 164425bec332c817387b87a72fecb906ef4678ed..0000000000000000000000000000000000000000 Binary files a/sgm/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/__pycache__/util.cpython-39.pyc b/sgm/__pycache__/util.cpython-39.pyc deleted file mode 100644 index f7e4c94b5bafb342981a65bb189bbaa3d2491342..0000000000000000000000000000000000000000 Binary files a/sgm/__pycache__/util.cpython-39.pyc and /dev/null differ diff --git a/sgm/models/__pycache__/__init__.cpython-39.pyc b/sgm/models/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 6011db5f39a94820a93ede8e0d10c17236d404ce..0000000000000000000000000000000000000000 Binary files a/sgm/models/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/models/__pycache__/autoencoder.cpython-39.pyc b/sgm/models/__pycache__/autoencoder.cpython-39.pyc deleted file mode 100644 index 176dd9d2c131e5440e0db6250e99cdd04ed8225b..0000000000000000000000000000000000000000 Binary files a/sgm/models/__pycache__/autoencoder.cpython-39.pyc and /dev/null differ diff --git a/sgm/models/__pycache__/diffusion.cpython-39.pyc b/sgm/models/__pycache__/diffusion.cpython-39.pyc deleted file mode 100644 index 65983da4a48b40d02534d4970e7ffceb7c927af0..0000000000000000000000000000000000000000 Binary files a/sgm/models/__pycache__/diffusion.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/__pycache__/__init__.cpython-39.pyc b/sgm/modules/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 721261e73684c84141bda627b91157ba5d0ecc28..0000000000000000000000000000000000000000 Binary files a/sgm/modules/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/__pycache__/attention.cpython-39.pyc b/sgm/modules/__pycache__/attention.cpython-39.pyc deleted file mode 100644 index 21763def6537a0166a2bd1ef1beda043d751932c..0000000000000000000000000000000000000000 Binary files a/sgm/modules/__pycache__/attention.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/__pycache__/ema.cpython-39.pyc b/sgm/modules/__pycache__/ema.cpython-39.pyc deleted file mode 100644 index 0f97aac98fc610c36d888cf8e5b64177d42cf5e2..0000000000000000000000000000000000000000 Binary files a/sgm/modules/__pycache__/ema.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/__pycache__/video_attention.cpython-39.pyc b/sgm/modules/__pycache__/video_attention.cpython-39.pyc deleted file mode 100644 index 7e7f1e82ec4bfdb45f99fd2e978b52856a3bd0ce..0000000000000000000000000000000000000000 Binary files a/sgm/modules/__pycache__/video_attention.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/autoencoding/__pycache__/__init__.cpython-39.pyc b/sgm/modules/autoencoding/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index eaf2bfca7b7785b3090fb24b276291a9754e14be..0000000000000000000000000000000000000000 Binary files a/sgm/modules/autoencoding/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/autoencoding/__pycache__/temporal_ae.cpython-39.pyc b/sgm/modules/autoencoding/__pycache__/temporal_ae.cpython-39.pyc deleted file mode 100644 index 2428596ecf918ece4be6c0f5259a18367f784369..0000000000000000000000000000000000000000 Binary files a/sgm/modules/autoencoding/__pycache__/temporal_ae.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/autoencoding/lpips/__pycache__/__init__.cpython-39.pyc b/sgm/modules/autoencoding/lpips/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index e6c5ccdfa4d4f8027fd00bb88d060b9328a3447b..0000000000000000000000000000000000000000 Binary files a/sgm/modules/autoencoding/lpips/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/autoencoding/lpips/__pycache__/util.cpython-39.pyc b/sgm/modules/autoencoding/lpips/__pycache__/util.cpython-39.pyc deleted file mode 100644 index fb078428aa63f32ff975a1178f3d88ff37989b84..0000000000000000000000000000000000000000 Binary files a/sgm/modules/autoencoding/lpips/__pycache__/util.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/autoencoding/lpips/loss/__pycache__/__init__.cpython-39.pyc b/sgm/modules/autoencoding/lpips/loss/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index ed8d29cbb2259e563654ecaed32f4f25378a210b..0000000000000000000000000000000000000000 Binary files a/sgm/modules/autoencoding/lpips/loss/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/autoencoding/lpips/loss/__pycache__/lpips.cpython-39.pyc b/sgm/modules/autoencoding/lpips/loss/__pycache__/lpips.cpython-39.pyc deleted file mode 100644 index 3fb860d74c38805d32597c4045829ab47dd63fa1..0000000000000000000000000000000000000000 Binary files a/sgm/modules/autoencoding/lpips/loss/__pycache__/lpips.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/autoencoding/regularizers/__pycache__/__init__.cpython-39.pyc b/sgm/modules/autoencoding/regularizers/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 1fed79c95b42eeecc1b332024666fd2c010b1349..0000000000000000000000000000000000000000 Binary files a/sgm/modules/autoencoding/regularizers/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/autoencoding/regularizers/__pycache__/base.cpython-39.pyc b/sgm/modules/autoencoding/regularizers/__pycache__/base.cpython-39.pyc deleted file mode 100644 index 3e43fb5e04c3b3762bad8fb1fad6bc57ef160e33..0000000000000000000000000000000000000000 Binary files a/sgm/modules/autoencoding/regularizers/__pycache__/base.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/diffusionmodules/__pycache__/__init__.cpython-39.pyc b/sgm/modules/diffusionmodules/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 7e3bfbd4106decc61630dee8aa86532f7512f79d..0000000000000000000000000000000000000000 Binary files a/sgm/modules/diffusionmodules/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/diffusionmodules/__pycache__/denoiser.cpython-39.pyc b/sgm/modules/diffusionmodules/__pycache__/denoiser.cpython-39.pyc deleted file mode 100644 index 356a880c134539fcc43b78098c26d69e5511f82a..0000000000000000000000000000000000000000 Binary files a/sgm/modules/diffusionmodules/__pycache__/denoiser.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/diffusionmodules/__pycache__/denoiser_scaling.cpython-39.pyc b/sgm/modules/diffusionmodules/__pycache__/denoiser_scaling.cpython-39.pyc deleted file mode 100644 index 516bc8049c03632769df0a3f5245a076711e5b75..0000000000000000000000000000000000000000 Binary files a/sgm/modules/diffusionmodules/__pycache__/denoiser_scaling.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/diffusionmodules/__pycache__/discretizer.cpython-39.pyc b/sgm/modules/diffusionmodules/__pycache__/discretizer.cpython-39.pyc deleted file mode 100644 index c88a3fea9e5c0f3d7e02d95155e553f54c0d5cd2..0000000000000000000000000000000000000000 Binary files a/sgm/modules/diffusionmodules/__pycache__/discretizer.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/diffusionmodules/__pycache__/loss.cpython-39.pyc b/sgm/modules/diffusionmodules/__pycache__/loss.cpython-39.pyc deleted file mode 100644 index d85dde1e520ed00dc25e81a3c4bfdb4116d0bf0f..0000000000000000000000000000000000000000 Binary files a/sgm/modules/diffusionmodules/__pycache__/loss.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/diffusionmodules/__pycache__/model.cpython-39.pyc b/sgm/modules/diffusionmodules/__pycache__/model.cpython-39.pyc deleted file mode 100644 index 1308643a1be41eb7ece36cb3bbc186d26222acf8..0000000000000000000000000000000000000000 Binary files a/sgm/modules/diffusionmodules/__pycache__/model.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/diffusionmodules/__pycache__/openaimodel.cpython-39.pyc b/sgm/modules/diffusionmodules/__pycache__/openaimodel.cpython-39.pyc deleted file mode 100644 index 0167d7f704590d68ef842f3da763969df260e56a..0000000000000000000000000000000000000000 Binary files a/sgm/modules/diffusionmodules/__pycache__/openaimodel.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/diffusionmodules/__pycache__/util.cpython-39.pyc b/sgm/modules/diffusionmodules/__pycache__/util.cpython-39.pyc deleted file mode 100644 index fd45cb704059baef5be4e6b8fc53af6ccddae29c..0000000000000000000000000000000000000000 Binary files a/sgm/modules/diffusionmodules/__pycache__/util.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/diffusionmodules/__pycache__/wrappers.cpython-39.pyc b/sgm/modules/diffusionmodules/__pycache__/wrappers.cpython-39.pyc deleted file mode 100644 index 23e7a997c27f7978d302053ae3032ba1387edae8..0000000000000000000000000000000000000000 Binary files a/sgm/modules/diffusionmodules/__pycache__/wrappers.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/distributions/__pycache__/__init__.cpython-39.pyc b/sgm/modules/distributions/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index fcfe026887b2724ac802661ebc90a2788da27543..0000000000000000000000000000000000000000 Binary files a/sgm/modules/distributions/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/distributions/__pycache__/distributions.cpython-39.pyc b/sgm/modules/distributions/__pycache__/distributions.cpython-39.pyc deleted file mode 100644 index b7f7e2ea7a2ec6426f332386d251a5f53ea6d46d..0000000000000000000000000000000000000000 Binary files a/sgm/modules/distributions/__pycache__/distributions.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/encoders/__pycache__/__init__.cpython-39.pyc b/sgm/modules/encoders/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 0d3fb8f72123d0c57b6ce7bdec76325d664ab315..0000000000000000000000000000000000000000 Binary files a/sgm/modules/encoders/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/sgm/modules/encoders/__pycache__/modules.cpython-39.pyc b/sgm/modules/encoders/__pycache__/modules.cpython-39.pyc deleted file mode 100644 index 172ce515f60549fa5e861bb0d29d6d979a306d66..0000000000000000000000000000000000000000 Binary files a/sgm/modules/encoders/__pycache__/modules.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/__pycache__/__init__.cpython-39.pyc b/torch_utils/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index b2c7935e1263357d879402cdef4272e878d34fc6..0000000000000000000000000000000000000000 Binary files a/torch_utils/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/__pycache__/components.cpython-39.pyc b/torch_utils/__pycache__/components.cpython-39.pyc deleted file mode 100644 index ca5b65878d611b1b78d4219688426c4f3e45776d..0000000000000000000000000000000000000000 Binary files a/torch_utils/__pycache__/components.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/__pycache__/custom_ops.cpython-39.pyc b/torch_utils/__pycache__/custom_ops.cpython-39.pyc deleted file mode 100644 index c302e49f8bad0a4f82befd0d1dddbc869832c612..0000000000000000000000000000000000000000 Binary files a/torch_utils/__pycache__/custom_ops.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/__pycache__/legacy.cpython-39.pyc b/torch_utils/__pycache__/legacy.cpython-39.pyc deleted file mode 100644 index a0e76f2c462bd10c8b3638d47657bd132b72ebb1..0000000000000000000000000000000000000000 Binary files a/torch_utils/__pycache__/legacy.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/__pycache__/misc.cpython-39.pyc b/torch_utils/__pycache__/misc.cpython-39.pyc deleted file mode 100644 index 1881a96d35d303182ce3805e2a1920eaf7a08dd7..0000000000000000000000000000000000000000 Binary files a/torch_utils/__pycache__/misc.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/__pycache__/persistence.cpython-39.pyc b/torch_utils/__pycache__/persistence.cpython-39.pyc deleted file mode 100644 index 7bb9f3dd151a5e6c4de98d72cba4de1f3058f9b2..0000000000000000000000000000000000000000 Binary files a/torch_utils/__pycache__/persistence.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/distributions/__pycache__/__init__.cpython-39.pyc b/torch_utils/distributions/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 8e78de3043b783d4bcb48bc9a04a7eb65243ed17..0000000000000000000000000000000000000000 Binary files a/torch_utils/distributions/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/distributions/__pycache__/distributions.cpython-39.pyc b/torch_utils/distributions/__pycache__/distributions.cpython-39.pyc deleted file mode 100644 index ddef35c3cabc41b394f47955362550ab4b0794ef..0000000000000000000000000000000000000000 Binary files a/torch_utils/distributions/__pycache__/distributions.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/ops/__pycache__/__init__.cpython-39.pyc b/torch_utils/ops/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 5fbfdd94279ac8b8aeb2e5e7df1caed6fef705df..0000000000000000000000000000000000000000 Binary files a/torch_utils/ops/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/ops/__pycache__/bias_act.cpython-39.pyc b/torch_utils/ops/__pycache__/bias_act.cpython-39.pyc deleted file mode 100644 index 5b41025bf242ab9e9ec2059663159f1b330144d2..0000000000000000000000000000000000000000 Binary files a/torch_utils/ops/__pycache__/bias_act.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/ops/__pycache__/conv2d_gradfix.cpython-39.pyc b/torch_utils/ops/__pycache__/conv2d_gradfix.cpython-39.pyc deleted file mode 100644 index 401735000e2c447afcdbaa50db329d118409241b..0000000000000000000000000000000000000000 Binary files a/torch_utils/ops/__pycache__/conv2d_gradfix.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/ops/__pycache__/conv2d_resample.cpython-39.pyc b/torch_utils/ops/__pycache__/conv2d_resample.cpython-39.pyc deleted file mode 100644 index 2ec280b11ab057b07f77bef8012ae50c9a039e72..0000000000000000000000000000000000000000 Binary files a/torch_utils/ops/__pycache__/conv2d_resample.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/ops/__pycache__/fma.cpython-39.pyc b/torch_utils/ops/__pycache__/fma.cpython-39.pyc deleted file mode 100644 index 78c6735106f424e42edc477540710983800f8052..0000000000000000000000000000000000000000 Binary files a/torch_utils/ops/__pycache__/fma.cpython-39.pyc and /dev/null differ diff --git a/torch_utils/ops/__pycache__/upfirdn2d.cpython-39.pyc b/torch_utils/ops/__pycache__/upfirdn2d.cpython-39.pyc deleted file mode 100644 index 32bc9166e3558337b6f561eb264c30b5f017b5e8..0000000000000000000000000000000000000000 Binary files a/torch_utils/ops/__pycache__/upfirdn2d.cpython-39.pyc and /dev/null differ diff --git a/transport/__pycache__/__init__.cpython-39.pyc b/transport/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index ea906abbf48c182e99c7dab1513f3fe38080427c..0000000000000000000000000000000000000000 Binary files a/transport/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/transport/__pycache__/integrators.cpython-39.pyc b/transport/__pycache__/integrators.cpython-39.pyc deleted file mode 100644 index 3d00db869cc4afd36930d533e5b9e6ac59438ef4..0000000000000000000000000000000000000000 Binary files a/transport/__pycache__/integrators.cpython-39.pyc and /dev/null differ diff --git a/transport/__pycache__/path.cpython-39.pyc b/transport/__pycache__/path.cpython-39.pyc deleted file mode 100644 index ec39463c0c54a31861667088dc4605a9023cb885..0000000000000000000000000000000000000000 Binary files a/transport/__pycache__/path.cpython-39.pyc and /dev/null differ diff --git a/transport/__pycache__/train_utils.cpython-39.pyc b/transport/__pycache__/train_utils.cpython-39.pyc deleted file mode 100644 index d9fbc447beb70ddb0f175987b3cbaae95515e283..0000000000000000000000000000000000000000 Binary files a/transport/__pycache__/train_utils.cpython-39.pyc and /dev/null differ diff --git a/transport/__pycache__/transport.cpython-39.pyc b/transport/__pycache__/transport.cpython-39.pyc deleted file mode 100644 index c031079d06c2ed53a9865a646ea5feeccc02a92c..0000000000000000000000000000000000000000 Binary files a/transport/__pycache__/transport.cpython-39.pyc and /dev/null differ diff --git a/transport/__pycache__/utils.cpython-39.pyc b/transport/__pycache__/utils.cpython-39.pyc deleted file mode 100644 index 35dbccc0bc006f0202d808b3441a3a9bc2dc6959..0000000000000000000000000000000000000000 Binary files a/transport/__pycache__/utils.cpython-39.pyc and /dev/null differ diff --git a/utils/__pycache__/__init__.cpython-39.pyc b/utils/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index 86210ce35ef47d6120f6b7d47079281e35e050db..0000000000000000000000000000000000000000 Binary files a/utils/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/utils/__pycache__/general_utils.cpython-39.pyc b/utils/__pycache__/general_utils.cpython-39.pyc deleted file mode 100644 index 124082a4705c2c3a704d017739564fc96232d6b9..0000000000000000000000000000000000000000 Binary files a/utils/__pycache__/general_utils.cpython-39.pyc and /dev/null differ diff --git a/utils/__pycache__/geometry.cpython-39.pyc b/utils/__pycache__/geometry.cpython-39.pyc deleted file mode 100644 index 9caabbe585cfe95325f55f57ec94f6d187fd5eb2..0000000000000000000000000000000000000000 Binary files a/utils/__pycache__/geometry.cpython-39.pyc and /dev/null differ diff --git a/utils/__pycache__/infer_utils.cpython-39.pyc b/utils/__pycache__/infer_utils.cpython-39.pyc deleted file mode 100644 index 241b1644e2def74b6d8027764234af74ced9f0d4..0000000000000000000000000000000000000000 Binary files a/utils/__pycache__/infer_utils.cpython-39.pyc and /dev/null differ diff --git a/utils/__pycache__/mesh_util.cpython-39.pyc b/utils/__pycache__/mesh_util.cpython-39.pyc deleted file mode 100644 index dda74763b99620f34b7dbcdf864d72c0138931bb..0000000000000000000000000000000000000000 Binary files a/utils/__pycache__/mesh_util.cpython-39.pyc and /dev/null differ diff --git a/utils/__pycache__/nerf_utils.cpython-39.pyc b/utils/__pycache__/nerf_utils.cpython-39.pyc deleted file mode 100644 index c567af2d0a8c1a712d48cd802b955f3c04ac1532..0000000000000000000000000000000000000000 Binary files a/utils/__pycache__/nerf_utils.cpython-39.pyc and /dev/null differ diff --git a/utils/__pycache__/point_utils.cpython-39.pyc b/utils/__pycache__/point_utils.cpython-39.pyc deleted file mode 100644 index 2b797aaae4399ba60702b4d14c690337d234a15e..0000000000000000000000000000000000000000 Binary files a/utils/__pycache__/point_utils.cpython-39.pyc and /dev/null differ diff --git a/utils/dust3r/__pycache__/__init__.cpython-39.pyc b/utils/dust3r/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index cf500f88c2bfd9dec420bdf999a0d83b86279da5..0000000000000000000000000000000000000000 Binary files a/utils/dust3r/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/utils/dust3r/__pycache__/dpt_block.cpython-39.pyc b/utils/dust3r/__pycache__/dpt_block.cpython-39.pyc deleted file mode 100644 index ccdd853204fa15ea98395877fa2e580d9ff71a5e..0000000000000000000000000000000000000000 Binary files a/utils/dust3r/__pycache__/dpt_block.cpython-39.pyc and /dev/null differ diff --git a/utils/dust3r/heads/__pycache__/__init__.cpython-39.pyc b/utils/dust3r/heads/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index aacea47d54e0ecc7907598569073c505a1bad0dc..0000000000000000000000000000000000000000 Binary files a/utils/dust3r/heads/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/utils/dust3r/heads/__pycache__/dpt_head.cpython-39.pyc b/utils/dust3r/heads/__pycache__/dpt_head.cpython-39.pyc deleted file mode 100644 index d3ee7c2b4d824aaad2dfdc9664f6c73f4a53a49c..0000000000000000000000000000000000000000 Binary files a/utils/dust3r/heads/__pycache__/dpt_head.cpython-39.pyc and /dev/null differ diff --git a/utils/dust3r/heads/__pycache__/linear_head.cpython-39.pyc b/utils/dust3r/heads/__pycache__/linear_head.cpython-39.pyc deleted file mode 100644 index e1f3628018549b618ebef97bb8554fee9d6b7805..0000000000000000000000000000000000000000 Binary files a/utils/dust3r/heads/__pycache__/linear_head.cpython-39.pyc and /dev/null differ diff --git a/utils/dust3r/heads/__pycache__/postprocess.cpython-39.pyc b/utils/dust3r/heads/__pycache__/postprocess.cpython-39.pyc deleted file mode 100644 index bbd79540a3222807c11a331971c5362b3a211082..0000000000000000000000000000000000000000 Binary files a/utils/dust3r/heads/__pycache__/postprocess.cpython-39.pyc and /dev/null differ diff --git a/utils/emd/__pycache__/emd_module.cpython-39.pyc b/utils/emd/__pycache__/emd_module.cpython-39.pyc deleted file mode 100644 index 57cb5851f4d396c21d280a4c5b9a22319db5f918..0000000000000000000000000000000000000000 Binary files a/utils/emd/__pycache__/emd_module.cpython-39.pyc and /dev/null differ diff --git a/utils/gs_utils/__pycache__/graphics_utils.cpython-39.pyc b/utils/gs_utils/__pycache__/graphics_utils.cpython-39.pyc deleted file mode 100644 index 1e8ccc71f3a814de8a49a4966b8fc61397aff835..0000000000000000000000000000000000000000 Binary files a/utils/gs_utils/__pycache__/graphics_utils.cpython-39.pyc and /dev/null differ diff --git a/vit/__pycache__/__init__.cpython-310.pyc b/vit/__pycache__/__init__.cpython-310.pyc deleted file mode 100644 index 80f5f83e44b87223725a5843acc84e6e3fc3ec45..0000000000000000000000000000000000000000 Binary files a/vit/__pycache__/__init__.cpython-310.pyc and /dev/null differ diff --git a/vit/__pycache__/__init__.cpython-39.pyc b/vit/__pycache__/__init__.cpython-39.pyc deleted file mode 100644 index ed79fba98ceec5875c876f875e17e513d1da3ff4..0000000000000000000000000000000000000000 Binary files a/vit/__pycache__/__init__.cpython-39.pyc and /dev/null differ diff --git a/vit/__pycache__/utils.cpython-310.pyc b/vit/__pycache__/utils.cpython-310.pyc deleted file mode 100644 index ef3b9b1244d9fd096321be781c9fab79a27aa48f..0000000000000000000000000000000000000000 Binary files a/vit/__pycache__/utils.cpython-310.pyc and /dev/null differ diff --git a/vit/__pycache__/utils.cpython-39.pyc b/vit/__pycache__/utils.cpython-39.pyc deleted file mode 100644 index 2378dede591ef27df493a03df3679d5f3b12e4c1..0000000000000000000000000000000000000000 Binary files a/vit/__pycache__/utils.cpython-39.pyc and /dev/null differ diff --git a/vit/__pycache__/vision_transformer.cpython-310.pyc b/vit/__pycache__/vision_transformer.cpython-310.pyc deleted file mode 100644 index f169e25054c58c1ac63dc37c85b00f2d3fe82862..0000000000000000000000000000000000000000 Binary files a/vit/__pycache__/vision_transformer.cpython-310.pyc and /dev/null differ diff --git a/vit/__pycache__/vision_transformer.cpython-39.pyc b/vit/__pycache__/vision_transformer.cpython-39.pyc deleted file mode 100644 index 17c1e0b12b2e0e2cf79fcc10495912686d7c024d..0000000000000000000000000000000000000000 Binary files a/vit/__pycache__/vision_transformer.cpython-39.pyc and /dev/null differ diff --git a/vit/__pycache__/vit_triplane.cpython-310.pyc b/vit/__pycache__/vit_triplane.cpython-310.pyc deleted file mode 100644 index ba73ed3fea971524c9596eb9c8ec3c825dc6eba6..0000000000000000000000000000000000000000 Binary files a/vit/__pycache__/vit_triplane.cpython-310.pyc and /dev/null differ diff --git a/vit/__pycache__/vit_triplane.cpython-39.pyc b/vit/__pycache__/vit_triplane.cpython-39.pyc deleted file mode 100644 index ea9125913b3dfdef0edb6ecfe6aada76d0241ed4..0000000000000000000000000000000000000000 Binary files a/vit/__pycache__/vit_triplane.cpython-39.pyc and /dev/null differ