├── .gitignore ├── LICENSE ├── README.md ├── blender ├── blender_lrm_script.py ├── distributed_uniform_lrm.py └── install_addon.py ├── canonicalize ├── __init__.py ├── models │ ├── attention.py │ ├── imageproj.py │ ├── refunet.py │ ├── resnet.py │ ├── transformer_mv2d.py │ ├── unet.py │ ├── unet_blocks.py │ ├── unet_mv2d_blocks.py │ ├── unet_mv2d_condition.py │ └── unet_mv2d_ref.py ├── pipeline_canonicalize.py └── util.py ├── configs ├── canonicalization-infer.yaml └── mesh-slrm-infer.yaml ├── data ├── test_list.json └── train_list.json ├── infer_canonicalize.py ├── infer_multiview.py ├── infer_refine.py ├── infer_slrm.py ├── input_cases ├── 1.png ├── 2.png ├── 3.png ├── 4.png ├── ayaka.png ├── belle.png └── firefly.png ├── multiview ├── __init__.py ├── fixed_prompt_embeds_6view │ ├── clr_embeds.pt │ └── normal_embeds.pt ├── models │ ├── transformer_mv2d_image.py │ ├── transformer_mv2d_rowwise.py │ ├── transformer_mv2d_self_rowwise.py │ ├── unet_mv2d_blocks.py │ └── unet_mv2d_condition.py └── pipeline_multiclass.py ├── refine ├── func.py ├── mesh_refine.py ├── opt.py ├── remesh.py └── render.py ├── requirements.txt └── slrm ├── __init__.py ├── cameras.npy ├── models ├── decoder │ ├── __init__.py │ └── transformer.py ├── encoder │ ├── __init__.py │ ├── dino.py │ └── dino_wrapper.py ├── geometry │ ├── __init__.py │ ├── camera │ │ ├── __init__.py │ │ └── perspective_camera.py │ ├── render │ │ ├── __init__.py │ │ └── neural_render.py │ └── rep_3d │ │ ├── __init__.py │ │ ├── dmtet.py │ │ ├── dmtet_utils.py │ │ ├── extract_texture_map.py │ │ ├── flexicubes.py │ │ ├── flexicubes_geometry.py │ │ └── tables.py ├── lrm.py ├── lrm_mesh.py └── renderer │ ├── __init__.py │ ├── synthesizer.py │ ├── synthesizer_mesh.py │ └── utils │ ├── __init__.py │ ├── math_utils.py │ ├── ray_marcher.py │ ├── ray_sampler.py │ └── renderer.py └── utils ├── camera_util.py ├── infer_util.py ├── mesh_util.py └── train_util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/README.md -------------------------------------------------------------------------------- /blender/blender_lrm_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/blender/blender_lrm_script.py -------------------------------------------------------------------------------- /blender/distributed_uniform_lrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/blender/distributed_uniform_lrm.py -------------------------------------------------------------------------------- /blender/install_addon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/blender/install_addon.py -------------------------------------------------------------------------------- /canonicalize/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /canonicalize/models/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/attention.py -------------------------------------------------------------------------------- /canonicalize/models/imageproj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/imageproj.py -------------------------------------------------------------------------------- /canonicalize/models/refunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/refunet.py -------------------------------------------------------------------------------- /canonicalize/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/resnet.py -------------------------------------------------------------------------------- /canonicalize/models/transformer_mv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/transformer_mv2d.py -------------------------------------------------------------------------------- /canonicalize/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/unet.py -------------------------------------------------------------------------------- /canonicalize/models/unet_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/unet_blocks.py -------------------------------------------------------------------------------- /canonicalize/models/unet_mv2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/unet_mv2d_blocks.py -------------------------------------------------------------------------------- /canonicalize/models/unet_mv2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/unet_mv2d_condition.py -------------------------------------------------------------------------------- /canonicalize/models/unet_mv2d_ref.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/models/unet_mv2d_ref.py -------------------------------------------------------------------------------- /canonicalize/pipeline_canonicalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/pipeline_canonicalize.py -------------------------------------------------------------------------------- /canonicalize/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/canonicalize/util.py -------------------------------------------------------------------------------- /configs/canonicalization-infer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/configs/canonicalization-infer.yaml -------------------------------------------------------------------------------- /configs/mesh-slrm-infer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/configs/mesh-slrm-infer.yaml -------------------------------------------------------------------------------- /data/test_list.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/data/test_list.json -------------------------------------------------------------------------------- /data/train_list.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/data/train_list.json -------------------------------------------------------------------------------- /infer_canonicalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/infer_canonicalize.py -------------------------------------------------------------------------------- /infer_multiview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/infer_multiview.py -------------------------------------------------------------------------------- /infer_refine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/infer_refine.py -------------------------------------------------------------------------------- /infer_slrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/infer_slrm.py -------------------------------------------------------------------------------- /input_cases/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/input_cases/1.png -------------------------------------------------------------------------------- /input_cases/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/input_cases/2.png -------------------------------------------------------------------------------- /input_cases/3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/input_cases/3.png -------------------------------------------------------------------------------- /input_cases/4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/input_cases/4.png -------------------------------------------------------------------------------- /input_cases/ayaka.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/input_cases/ayaka.png -------------------------------------------------------------------------------- /input_cases/belle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/input_cases/belle.png -------------------------------------------------------------------------------- /input_cases/firefly.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/input_cases/firefly.png -------------------------------------------------------------------------------- /multiview/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multiview/fixed_prompt_embeds_6view/clr_embeds.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/multiview/fixed_prompt_embeds_6view/clr_embeds.pt -------------------------------------------------------------------------------- /multiview/fixed_prompt_embeds_6view/normal_embeds.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/multiview/fixed_prompt_embeds_6view/normal_embeds.pt -------------------------------------------------------------------------------- /multiview/models/transformer_mv2d_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/multiview/models/transformer_mv2d_image.py -------------------------------------------------------------------------------- /multiview/models/transformer_mv2d_rowwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/multiview/models/transformer_mv2d_rowwise.py -------------------------------------------------------------------------------- /multiview/models/transformer_mv2d_self_rowwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/multiview/models/transformer_mv2d_self_rowwise.py -------------------------------------------------------------------------------- /multiview/models/unet_mv2d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/multiview/models/unet_mv2d_blocks.py -------------------------------------------------------------------------------- /multiview/models/unet_mv2d_condition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/multiview/models/unet_mv2d_condition.py -------------------------------------------------------------------------------- /multiview/pipeline_multiclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/multiview/pipeline_multiclass.py -------------------------------------------------------------------------------- /refine/func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/refine/func.py -------------------------------------------------------------------------------- /refine/mesh_refine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/refine/mesh_refine.py -------------------------------------------------------------------------------- /refine/opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/refine/opt.py -------------------------------------------------------------------------------- /refine/remesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/refine/remesh.py -------------------------------------------------------------------------------- /refine/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/refine/render.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/requirements.txt -------------------------------------------------------------------------------- /slrm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /slrm/cameras.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/cameras.npy -------------------------------------------------------------------------------- /slrm/models/decoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /slrm/models/decoder/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/decoder/transformer.py -------------------------------------------------------------------------------- /slrm/models/encoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /slrm/models/encoder/dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/encoder/dino.py -------------------------------------------------------------------------------- /slrm/models/encoder/dino_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/encoder/dino_wrapper.py -------------------------------------------------------------------------------- /slrm/models/geometry/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/__init__.py -------------------------------------------------------------------------------- /slrm/models/geometry/camera/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/camera/__init__.py -------------------------------------------------------------------------------- /slrm/models/geometry/camera/perspective_camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/camera/perspective_camera.py -------------------------------------------------------------------------------- /slrm/models/geometry/render/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/render/__init__.py -------------------------------------------------------------------------------- /slrm/models/geometry/render/neural_render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/render/neural_render.py -------------------------------------------------------------------------------- /slrm/models/geometry/rep_3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/rep_3d/__init__.py -------------------------------------------------------------------------------- /slrm/models/geometry/rep_3d/dmtet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/rep_3d/dmtet.py -------------------------------------------------------------------------------- /slrm/models/geometry/rep_3d/dmtet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/rep_3d/dmtet_utils.py -------------------------------------------------------------------------------- /slrm/models/geometry/rep_3d/extract_texture_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/rep_3d/extract_texture_map.py -------------------------------------------------------------------------------- /slrm/models/geometry/rep_3d/flexicubes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/rep_3d/flexicubes.py -------------------------------------------------------------------------------- /slrm/models/geometry/rep_3d/flexicubes_geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/rep_3d/flexicubes_geometry.py -------------------------------------------------------------------------------- /slrm/models/geometry/rep_3d/tables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/geometry/rep_3d/tables.py -------------------------------------------------------------------------------- /slrm/models/lrm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/lrm.py -------------------------------------------------------------------------------- /slrm/models/lrm_mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/lrm_mesh.py -------------------------------------------------------------------------------- /slrm/models/renderer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/renderer/__init__.py -------------------------------------------------------------------------------- /slrm/models/renderer/synthesizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/renderer/synthesizer.py -------------------------------------------------------------------------------- /slrm/models/renderer/synthesizer_mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/renderer/synthesizer_mesh.py -------------------------------------------------------------------------------- /slrm/models/renderer/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/renderer/utils/__init__.py -------------------------------------------------------------------------------- /slrm/models/renderer/utils/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/renderer/utils/math_utils.py -------------------------------------------------------------------------------- /slrm/models/renderer/utils/ray_marcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/renderer/utils/ray_marcher.py -------------------------------------------------------------------------------- /slrm/models/renderer/utils/ray_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/renderer/utils/ray_sampler.py -------------------------------------------------------------------------------- /slrm/models/renderer/utils/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/models/renderer/utils/renderer.py -------------------------------------------------------------------------------- /slrm/utils/camera_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/utils/camera_util.py -------------------------------------------------------------------------------- /slrm/utils/infer_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/utils/infer_util.py -------------------------------------------------------------------------------- /slrm/utils/mesh_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/utils/mesh_util.py -------------------------------------------------------------------------------- /slrm/utils/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hyz317/StdGEN/HEAD/slrm/utils/train_util.py --------------------------------------------------------------------------------