├── .env.example ├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── assets ├── axis.png ├── cast.drawio ├── cast_demo_tripo.png ├── cast_demo_workflow.png ├── init_rot_matrices.npy ├── inpaint_failure.png ├── inputs │ ├── bicycle.png │ ├── bicycle2.png │ ├── book1.png │ ├── book2.png │ ├── box.png │ ├── car1.png │ ├── car2.png │ ├── doll.png │ ├── doll2.png │ ├── image1.png │ ├── image2.png │ ├── image3.png │ ├── indoor.png │ ├── outdoor.png │ └── starbuck.png ├── nvdrexp.jpg └── teaser.png ├── cast ├── __init__.py ├── __main__.py ├── cli.py ├── config │ ├── __init__.py │ └── settings.py ├── core │ ├── __init__.py │ ├── common.py │ └── pipeline.py ├── modules │ ├── __init__.py │ ├── depth_estimation.py │ ├── detection_filtering.py │ ├── detection_segmentation.py │ ├── image_generation.py │ ├── mesh_generation.py │ ├── pose_estimation.py │ ├── pose_optimizer.py │ ├── render_compare.py │ └── scene_graph_optimization.py └── utils │ ├── __init__.py │ ├── api_clients.py │ ├── image_utils.py │ ├── open3d_sdf.py │ └── torchvision_fix.py ├── demo.bat ├── install_setup.py ├── requirements.txt └── setup.py /.env.example: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/.env.example -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/README.md -------------------------------------------------------------------------------- /assets/axis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/axis.png -------------------------------------------------------------------------------- /assets/cast.drawio: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/cast.drawio -------------------------------------------------------------------------------- /assets/cast_demo_tripo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/cast_demo_tripo.png -------------------------------------------------------------------------------- /assets/cast_demo_workflow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/cast_demo_workflow.png -------------------------------------------------------------------------------- /assets/init_rot_matrices.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/init_rot_matrices.npy -------------------------------------------------------------------------------- /assets/inpaint_failure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inpaint_failure.png -------------------------------------------------------------------------------- /assets/inputs/bicycle.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/bicycle.png -------------------------------------------------------------------------------- /assets/inputs/bicycle2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/bicycle2.png -------------------------------------------------------------------------------- /assets/inputs/book1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/book1.png -------------------------------------------------------------------------------- /assets/inputs/book2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/book2.png -------------------------------------------------------------------------------- /assets/inputs/box.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/box.png -------------------------------------------------------------------------------- /assets/inputs/car1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/car1.png -------------------------------------------------------------------------------- /assets/inputs/car2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/car2.png -------------------------------------------------------------------------------- /assets/inputs/doll.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/doll.png -------------------------------------------------------------------------------- /assets/inputs/doll2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/doll2.png -------------------------------------------------------------------------------- /assets/inputs/image1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/image1.png -------------------------------------------------------------------------------- /assets/inputs/image2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/image2.png -------------------------------------------------------------------------------- /assets/inputs/image3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/image3.png -------------------------------------------------------------------------------- /assets/inputs/indoor.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/indoor.png -------------------------------------------------------------------------------- /assets/inputs/outdoor.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/outdoor.png -------------------------------------------------------------------------------- /assets/inputs/starbuck.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/inputs/starbuck.png -------------------------------------------------------------------------------- /assets/nvdrexp.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/nvdrexp.jpg -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /cast/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/__init__.py -------------------------------------------------------------------------------- /cast/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/__main__.py -------------------------------------------------------------------------------- /cast/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/cli.py -------------------------------------------------------------------------------- /cast/config/__init__.py: -------------------------------------------------------------------------------- 1 | """Configuration management""" -------------------------------------------------------------------------------- /cast/config/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/config/settings.py -------------------------------------------------------------------------------- /cast/core/__init__.py: -------------------------------------------------------------------------------- 1 | """Core pipeline components""" -------------------------------------------------------------------------------- /cast/core/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/core/common.py -------------------------------------------------------------------------------- /cast/core/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/core/pipeline.py -------------------------------------------------------------------------------- /cast/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/__init__.py -------------------------------------------------------------------------------- /cast/modules/depth_estimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/depth_estimation.py -------------------------------------------------------------------------------- /cast/modules/detection_filtering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/detection_filtering.py -------------------------------------------------------------------------------- /cast/modules/detection_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/detection_segmentation.py -------------------------------------------------------------------------------- /cast/modules/image_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/image_generation.py -------------------------------------------------------------------------------- /cast/modules/mesh_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/mesh_generation.py -------------------------------------------------------------------------------- /cast/modules/pose_estimation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/pose_estimation.py -------------------------------------------------------------------------------- /cast/modules/pose_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/pose_optimizer.py -------------------------------------------------------------------------------- /cast/modules/render_compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/render_compare.py -------------------------------------------------------------------------------- /cast/modules/scene_graph_optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/modules/scene_graph_optimization.py -------------------------------------------------------------------------------- /cast/utils/__init__.py: -------------------------------------------------------------------------------- 1 | """Utility functions and helpers""" -------------------------------------------------------------------------------- /cast/utils/api_clients.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/utils/api_clients.py -------------------------------------------------------------------------------- /cast/utils/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/utils/image_utils.py -------------------------------------------------------------------------------- /cast/utils/open3d_sdf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/utils/open3d_sdf.py -------------------------------------------------------------------------------- /cast/utils/torchvision_fix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/cast/utils/torchvision_fix.py -------------------------------------------------------------------------------- /demo.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/demo.bat -------------------------------------------------------------------------------- /install_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/install_setup.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FishWoWater/CAST/HEAD/setup.py --------------------------------------------------------------------------------