├── .gitignore ├── README.md ├── evaluate_diffusion.py ├── evaluate_normalized_metric.py ├── evaluate_upsampler.py ├── evaluate_vae.py ├── example ├── dataset │ ├── .DS_Store │ ├── images │ │ ├── 0a0bc2921e5246a28732bf5584c251d1 │ │ │ ├── 000.png │ │ │ ├── 001.png │ │ │ ├── 002.png │ │ │ └── 003.png │ │ ├── 0a0bec73aeb9466e875ae760fd75e64a │ │ │ └── 000.png │ │ └── 0a0c75dab0844e7fa5b299d4af858bec │ │ │ ├── 000.png │ │ │ ├── 001.png │ │ │ ├── 002.png │ │ │ └── 003.png │ └── xrays │ │ ├── 0a0bc2921e5246a28732bf5584c251d1 │ │ ├── 000.npz │ │ ├── 001.npz │ │ ├── 002.npz │ │ └── 003.npz │ │ ├── 0a0bec73aeb9466e875ae760fd75e64a │ │ └── 000.npz │ │ └── 0a0c75dab0844e7fa5b299d4af858bec │ │ ├── 000.npz │ │ ├── 001.npz │ │ ├── 002.npz │ │ └── 003.npz └── meshes │ └── 10716a366de708b8fac96522b26f7fd │ ├── images │ └── texture0.png │ └── models │ ├── 10716a366de708b8fac96522b26f7fd.mtl │ └── 10716a366de708b8fac96522b26f7fd.obj ├── inference_hr.py ├── inference_lr.py ├── preprocess ├── get_image │ ├── blender │ │ ├── blender_script.py │ │ └── distributed.py │ ├── custom │ │ ├── mesh_list.json │ │ └── render_mesh.sh │ └── objaverse │ │ ├── objaverse_80k.json │ │ └── render_mesh.sh └── get_xray │ ├── gen_gso.py │ ├── gen_objaverse.py │ └── gen_shapenet.py ├── requirements.txt ├── scripts ├── check_data.py ├── check_part.py ├── evaluation_gso.sh ├── evaluation_objaverse.sh ├── evaluation_shapenet.sh ├── evaluation_shapenet_casualvae.sh ├── evaluation_vae.sh ├── filter_dataset_by_iou.py ├── train_casualvae.sh ├── train_diffusion.sh ├── train_shapenet.sh ├── train_upsampler.sh ├── train_vae.sh └── upload.sh ├── src ├── __init__.py ├── chamfer_distance.py ├── dataset.py ├── metrics.py ├── utils.py ├── xray_decoder.json ├── xray_decoder.py ├── xray_decoder_large.json ├── xray_pipeline.py └── xray_vae.json ├── train_diffusion.py ├── train_upsampler.py └── train_vae.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/README.md -------------------------------------------------------------------------------- /evaluate_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/evaluate_diffusion.py -------------------------------------------------------------------------------- /evaluate_normalized_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/evaluate_normalized_metric.py -------------------------------------------------------------------------------- /evaluate_upsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/evaluate_upsampler.py -------------------------------------------------------------------------------- /evaluate_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/evaluate_vae.py -------------------------------------------------------------------------------- /example/dataset/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/.DS_Store -------------------------------------------------------------------------------- /example/dataset/images/0a0bc2921e5246a28732bf5584c251d1/000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/images/0a0bc2921e5246a28732bf5584c251d1/000.png -------------------------------------------------------------------------------- /example/dataset/images/0a0bc2921e5246a28732bf5584c251d1/001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/images/0a0bc2921e5246a28732bf5584c251d1/001.png -------------------------------------------------------------------------------- /example/dataset/images/0a0bc2921e5246a28732bf5584c251d1/002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/images/0a0bc2921e5246a28732bf5584c251d1/002.png -------------------------------------------------------------------------------- /example/dataset/images/0a0bc2921e5246a28732bf5584c251d1/003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/images/0a0bc2921e5246a28732bf5584c251d1/003.png -------------------------------------------------------------------------------- /example/dataset/images/0a0bec73aeb9466e875ae760fd75e64a/000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/images/0a0bec73aeb9466e875ae760fd75e64a/000.png -------------------------------------------------------------------------------- /example/dataset/images/0a0c75dab0844e7fa5b299d4af858bec/000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/images/0a0c75dab0844e7fa5b299d4af858bec/000.png -------------------------------------------------------------------------------- /example/dataset/images/0a0c75dab0844e7fa5b299d4af858bec/001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/images/0a0c75dab0844e7fa5b299d4af858bec/001.png -------------------------------------------------------------------------------- /example/dataset/images/0a0c75dab0844e7fa5b299d4af858bec/002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/images/0a0c75dab0844e7fa5b299d4af858bec/002.png -------------------------------------------------------------------------------- /example/dataset/images/0a0c75dab0844e7fa5b299d4af858bec/003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/images/0a0c75dab0844e7fa5b299d4af858bec/003.png -------------------------------------------------------------------------------- /example/dataset/xrays/0a0bc2921e5246a28732bf5584c251d1/000.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/xrays/0a0bc2921e5246a28732bf5584c251d1/000.npz -------------------------------------------------------------------------------- /example/dataset/xrays/0a0bc2921e5246a28732bf5584c251d1/001.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/xrays/0a0bc2921e5246a28732bf5584c251d1/001.npz -------------------------------------------------------------------------------- /example/dataset/xrays/0a0bc2921e5246a28732bf5584c251d1/002.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/xrays/0a0bc2921e5246a28732bf5584c251d1/002.npz -------------------------------------------------------------------------------- /example/dataset/xrays/0a0bc2921e5246a28732bf5584c251d1/003.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/xrays/0a0bc2921e5246a28732bf5584c251d1/003.npz -------------------------------------------------------------------------------- /example/dataset/xrays/0a0bec73aeb9466e875ae760fd75e64a/000.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/xrays/0a0bec73aeb9466e875ae760fd75e64a/000.npz -------------------------------------------------------------------------------- /example/dataset/xrays/0a0c75dab0844e7fa5b299d4af858bec/000.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/xrays/0a0c75dab0844e7fa5b299d4af858bec/000.npz -------------------------------------------------------------------------------- /example/dataset/xrays/0a0c75dab0844e7fa5b299d4af858bec/001.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/xrays/0a0c75dab0844e7fa5b299d4af858bec/001.npz -------------------------------------------------------------------------------- /example/dataset/xrays/0a0c75dab0844e7fa5b299d4af858bec/002.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/xrays/0a0c75dab0844e7fa5b299d4af858bec/002.npz -------------------------------------------------------------------------------- /example/dataset/xrays/0a0c75dab0844e7fa5b299d4af858bec/003.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/dataset/xrays/0a0c75dab0844e7fa5b299d4af858bec/003.npz -------------------------------------------------------------------------------- /example/meshes/10716a366de708b8fac96522b26f7fd/images/texture0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/meshes/10716a366de708b8fac96522b26f7fd/images/texture0.png -------------------------------------------------------------------------------- /example/meshes/10716a366de708b8fac96522b26f7fd/models/10716a366de708b8fac96522b26f7fd.mtl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/meshes/10716a366de708b8fac96522b26f7fd/models/10716a366de708b8fac96522b26f7fd.mtl -------------------------------------------------------------------------------- /example/meshes/10716a366de708b8fac96522b26f7fd/models/10716a366de708b8fac96522b26f7fd.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/example/meshes/10716a366de708b8fac96522b26f7fd/models/10716a366de708b8fac96522b26f7fd.obj -------------------------------------------------------------------------------- /inference_hr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/inference_hr.py -------------------------------------------------------------------------------- /inference_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/inference_lr.py -------------------------------------------------------------------------------- /preprocess/get_image/blender/blender_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/preprocess/get_image/blender/blender_script.py -------------------------------------------------------------------------------- /preprocess/get_image/blender/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/preprocess/get_image/blender/distributed.py -------------------------------------------------------------------------------- /preprocess/get_image/custom/mesh_list.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/preprocess/get_image/custom/mesh_list.json -------------------------------------------------------------------------------- /preprocess/get_image/custom/render_mesh.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/preprocess/get_image/custom/render_mesh.sh -------------------------------------------------------------------------------- /preprocess/get_image/objaverse/objaverse_80k.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/preprocess/get_image/objaverse/objaverse_80k.json -------------------------------------------------------------------------------- /preprocess/get_image/objaverse/render_mesh.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/preprocess/get_image/objaverse/render_mesh.sh -------------------------------------------------------------------------------- /preprocess/get_xray/gen_gso.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/preprocess/get_xray/gen_gso.py -------------------------------------------------------------------------------- /preprocess/get_xray/gen_objaverse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/preprocess/get_xray/gen_objaverse.py -------------------------------------------------------------------------------- /preprocess/get_xray/gen_shapenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/preprocess/get_xray/gen_shapenet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/check_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/check_data.py -------------------------------------------------------------------------------- /scripts/check_part.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/check_part.py -------------------------------------------------------------------------------- /scripts/evaluation_gso.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/evaluation_gso.sh -------------------------------------------------------------------------------- /scripts/evaluation_objaverse.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/evaluation_objaverse.sh -------------------------------------------------------------------------------- /scripts/evaluation_shapenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/evaluation_shapenet.sh -------------------------------------------------------------------------------- /scripts/evaluation_shapenet_casualvae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/evaluation_shapenet_casualvae.sh -------------------------------------------------------------------------------- /scripts/evaluation_vae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/evaluation_vae.sh -------------------------------------------------------------------------------- /scripts/filter_dataset_by_iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/filter_dataset_by_iou.py -------------------------------------------------------------------------------- /scripts/train_casualvae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/train_casualvae.sh -------------------------------------------------------------------------------- /scripts/train_diffusion.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/train_diffusion.sh -------------------------------------------------------------------------------- /scripts/train_shapenet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/train_shapenet.sh -------------------------------------------------------------------------------- /scripts/train_upsampler.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/train_upsampler.sh -------------------------------------------------------------------------------- /scripts/train_vae.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/train_vae.sh -------------------------------------------------------------------------------- /scripts/upload.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/scripts/upload.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/chamfer_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/src/chamfer_distance.py -------------------------------------------------------------------------------- /src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/src/dataset.py -------------------------------------------------------------------------------- /src/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/src/metrics.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/xray_decoder.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/src/xray_decoder.json -------------------------------------------------------------------------------- /src/xray_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/src/xray_decoder.py -------------------------------------------------------------------------------- /src/xray_decoder_large.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/src/xray_decoder_large.json -------------------------------------------------------------------------------- /src/xray_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/src/xray_pipeline.py -------------------------------------------------------------------------------- /src/xray_vae.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/src/xray_vae.json -------------------------------------------------------------------------------- /train_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/train_diffusion.py -------------------------------------------------------------------------------- /train_upsampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/train_upsampler.py -------------------------------------------------------------------------------- /train_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tau-yihouxiang/X-Ray/HEAD/train_vae.py --------------------------------------------------------------------------------