├── .gitignore ├── .gitmodules ├── LICENSE.txt ├── README.md ├── dataloader_rs ├── .gitignore ├── Cargo.lock ├── Cargo.toml └── src │ ├── bin │ ├── mixsvbrdfs.rs │ └── renderall.rs │ ├── form.rs │ ├── gen.rs │ ├── ids.rs │ ├── lib.rs │ ├── loaders.rs │ └── warp.rs ├── datasets ├── real_test_lookahead.yml ├── real_test_materialgan.yml ├── test_rasterized.yml ├── test_rendered_sunny_env.yml ├── test_rendered_sunny_fnf.yml ├── train_rasterized.yml ├── train_rendered_env.yml └── train_rendered_fnf.yml ├── demo ├── pink_input.png ├── pink_svbrdf.png ├── wildflower_sign_input.png └── wildflower_sign_svbrdf.png ├── environment.yml ├── eval.py ├── matfusion_diffusers_env_demo.ipynb ├── matfusion_diffusers_flash_demo.ipynb ├── matfusion_jax ├── config.py ├── data.so ├── model.py ├── net │ ├── mine.py │ └── resnet.py ├── nprast.py ├── pipeline.py └── vis.py ├── matfusion_jax_flash_demo.ipynb ├── score.py ├── scripts ├── compress_exrs.jl ├── convert_inria.py ├── convert_to_diffusers.py ├── integrate-normals.jl ├── prune_checkpoint.py └── render_synthetic │ ├── .gitignore │ ├── nogi_render_template.blend │ ├── ortho_render_control.py │ ├── ortho_render_template.blend │ ├── render_control.py │ └── render_template.blend ├── train.py └── view_eval.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/README.md -------------------------------------------------------------------------------- /dataloader_rs/.gitignore: -------------------------------------------------------------------------------- 1 | target/ 2 | -------------------------------------------------------------------------------- /dataloader_rs/Cargo.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/Cargo.lock -------------------------------------------------------------------------------- /dataloader_rs/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/Cargo.toml -------------------------------------------------------------------------------- /dataloader_rs/src/bin/mixsvbrdfs.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/src/bin/mixsvbrdfs.rs -------------------------------------------------------------------------------- /dataloader_rs/src/bin/renderall.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/src/bin/renderall.rs -------------------------------------------------------------------------------- /dataloader_rs/src/form.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/src/form.rs -------------------------------------------------------------------------------- /dataloader_rs/src/gen.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/src/gen.rs -------------------------------------------------------------------------------- /dataloader_rs/src/ids.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/src/ids.rs -------------------------------------------------------------------------------- /dataloader_rs/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/src/lib.rs -------------------------------------------------------------------------------- /dataloader_rs/src/loaders.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/src/loaders.rs -------------------------------------------------------------------------------- /dataloader_rs/src/warp.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/dataloader_rs/src/warp.rs -------------------------------------------------------------------------------- /datasets/real_test_lookahead.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/datasets/real_test_lookahead.yml -------------------------------------------------------------------------------- /datasets/real_test_materialgan.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/datasets/real_test_materialgan.yml -------------------------------------------------------------------------------- /datasets/test_rasterized.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/datasets/test_rasterized.yml -------------------------------------------------------------------------------- /datasets/test_rendered_sunny_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/datasets/test_rendered_sunny_env.yml -------------------------------------------------------------------------------- /datasets/test_rendered_sunny_fnf.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/datasets/test_rendered_sunny_fnf.yml -------------------------------------------------------------------------------- /datasets/train_rasterized.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/datasets/train_rasterized.yml -------------------------------------------------------------------------------- /datasets/train_rendered_env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/datasets/train_rendered_env.yml -------------------------------------------------------------------------------- /datasets/train_rendered_fnf.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/datasets/train_rendered_fnf.yml -------------------------------------------------------------------------------- /demo/pink_input.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/demo/pink_input.png -------------------------------------------------------------------------------- /demo/pink_svbrdf.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/demo/pink_svbrdf.png -------------------------------------------------------------------------------- /demo/wildflower_sign_input.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/demo/wildflower_sign_input.png -------------------------------------------------------------------------------- /demo/wildflower_sign_svbrdf.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/demo/wildflower_sign_svbrdf.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/environment.yml -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/eval.py -------------------------------------------------------------------------------- /matfusion_diffusers_env_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_diffusers_env_demo.ipynb -------------------------------------------------------------------------------- /matfusion_diffusers_flash_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_diffusers_flash_demo.ipynb -------------------------------------------------------------------------------- /matfusion_jax/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_jax/config.py -------------------------------------------------------------------------------- /matfusion_jax/data.so: -------------------------------------------------------------------------------- 1 | ../dataloader_rs/target/release/libdata.so -------------------------------------------------------------------------------- /matfusion_jax/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_jax/model.py -------------------------------------------------------------------------------- /matfusion_jax/net/mine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_jax/net/mine.py -------------------------------------------------------------------------------- /matfusion_jax/net/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_jax/net/resnet.py -------------------------------------------------------------------------------- /matfusion_jax/nprast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_jax/nprast.py -------------------------------------------------------------------------------- /matfusion_jax/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_jax/pipeline.py -------------------------------------------------------------------------------- /matfusion_jax/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_jax/vis.py -------------------------------------------------------------------------------- /matfusion_jax_flash_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/matfusion_jax_flash_demo.ipynb -------------------------------------------------------------------------------- /score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/score.py -------------------------------------------------------------------------------- /scripts/compress_exrs.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/compress_exrs.jl -------------------------------------------------------------------------------- /scripts/convert_inria.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/convert_inria.py -------------------------------------------------------------------------------- /scripts/convert_to_diffusers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/convert_to_diffusers.py -------------------------------------------------------------------------------- /scripts/integrate-normals.jl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/integrate-normals.jl -------------------------------------------------------------------------------- /scripts/prune_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/prune_checkpoint.py -------------------------------------------------------------------------------- /scripts/render_synthetic/.gitignore: -------------------------------------------------------------------------------- 1 | *.blend1 2 | data 3 | __pycache__ 4 | svbrdf-renders/ 5 | -------------------------------------------------------------------------------- /scripts/render_synthetic/nogi_render_template.blend: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/render_synthetic/nogi_render_template.blend -------------------------------------------------------------------------------- /scripts/render_synthetic/ortho_render_control.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/render_synthetic/ortho_render_control.py -------------------------------------------------------------------------------- /scripts/render_synthetic/ortho_render_template.blend: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/render_synthetic/ortho_render_template.blend -------------------------------------------------------------------------------- /scripts/render_synthetic/render_control.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/render_synthetic/render_control.py -------------------------------------------------------------------------------- /scripts/render_synthetic/render_template.blend: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/scripts/render_synthetic/render_template.blend -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/train.py -------------------------------------------------------------------------------- /view_eval.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samsartor/matfusion/HEAD/view_eval.ipynb --------------------------------------------------------------------------------