├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── app.py ├── assets ├── chair-8-diet.gif ├── chair.png ├── comcom.jpeg ├── drawing.png ├── drum_.png ├── hotdog_.png ├── lego-14-occ-diet_.gif ├── lego-14-occ-nerf_.gif ├── lego-8-diet-image.png ├── lego-8-diet.gif ├── lego-8-nerf.gif ├── materials-8-diet-image.png ├── mic-8-diet.gif ├── report_thumbnail.png ├── ship-8-diet-coarse.gif ├── ship-8-nerf.gif ├── ship-dietnerf.gif ├── ship-nerf.gif ├── ship_.png └── space_demo.png ├── configs ├── diet_nerf_tpu_vm_4shot.yaml ├── diet_nerf_tpu_vm_few_shot.yaml ├── diet_nerf_tpu_vm_test.yaml ├── eval_diet_nerf_tpu_vm_few_shot.yaml ├── nerf_tpu_vm_4shot.yaml └── nerf_tpu_vm_few_shot.yaml ├── demo ├── __init__.py ├── config.json └── src │ ├── __init__.py │ ├── config.py │ ├── models.py │ └── utils.py ├── eval.py ├── eval.sh ├── nerf ├── __init__.py ├── clip_utils.py ├── datasets.py ├── model_utils.py ├── models.py └── utils.py ├── requirements.txt ├── requirements_demo.txt ├── run.sh ├── train.py └── train.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/__init__.py -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/app.py -------------------------------------------------------------------------------- /assets/chair-8-diet.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/chair-8-diet.gif -------------------------------------------------------------------------------- /assets/chair.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/chair.png -------------------------------------------------------------------------------- /assets/comcom.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/comcom.jpeg -------------------------------------------------------------------------------- /assets/drawing.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/drawing.png -------------------------------------------------------------------------------- /assets/drum_.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/drum_.png -------------------------------------------------------------------------------- /assets/hotdog_.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/hotdog_.png -------------------------------------------------------------------------------- /assets/lego-14-occ-diet_.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/lego-14-occ-diet_.gif -------------------------------------------------------------------------------- /assets/lego-14-occ-nerf_.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/lego-14-occ-nerf_.gif -------------------------------------------------------------------------------- /assets/lego-8-diet-image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/lego-8-diet-image.png -------------------------------------------------------------------------------- /assets/lego-8-diet.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/lego-8-diet.gif -------------------------------------------------------------------------------- /assets/lego-8-nerf.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/lego-8-nerf.gif -------------------------------------------------------------------------------- /assets/materials-8-diet-image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/materials-8-diet-image.png -------------------------------------------------------------------------------- /assets/mic-8-diet.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/mic-8-diet.gif -------------------------------------------------------------------------------- /assets/report_thumbnail.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/report_thumbnail.png -------------------------------------------------------------------------------- /assets/ship-8-diet-coarse.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/ship-8-diet-coarse.gif -------------------------------------------------------------------------------- /assets/ship-8-nerf.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/ship-8-nerf.gif -------------------------------------------------------------------------------- /assets/ship-dietnerf.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/ship-dietnerf.gif -------------------------------------------------------------------------------- /assets/ship-nerf.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/ship-nerf.gif -------------------------------------------------------------------------------- /assets/ship_.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/ship_.png -------------------------------------------------------------------------------- /assets/space_demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/assets/space_demo.png -------------------------------------------------------------------------------- /configs/diet_nerf_tpu_vm_4shot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/configs/diet_nerf_tpu_vm_4shot.yaml -------------------------------------------------------------------------------- /configs/diet_nerf_tpu_vm_few_shot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/configs/diet_nerf_tpu_vm_few_shot.yaml -------------------------------------------------------------------------------- /configs/diet_nerf_tpu_vm_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/configs/diet_nerf_tpu_vm_test.yaml -------------------------------------------------------------------------------- /configs/eval_diet_nerf_tpu_vm_few_shot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/configs/eval_diet_nerf_tpu_vm_few_shot.yaml -------------------------------------------------------------------------------- /configs/nerf_tpu_vm_4shot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/configs/nerf_tpu_vm_4shot.yaml -------------------------------------------------------------------------------- /configs/nerf_tpu_vm_few_shot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/configs/nerf_tpu_vm_few_shot.yaml -------------------------------------------------------------------------------- /demo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /demo/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/demo/config.json -------------------------------------------------------------------------------- /demo/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /demo/src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/demo/src/config.py -------------------------------------------------------------------------------- /demo/src/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/demo/src/models.py -------------------------------------------------------------------------------- /demo/src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/demo/src/utils.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/eval.py -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/eval.sh -------------------------------------------------------------------------------- /nerf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/nerf/__init__.py -------------------------------------------------------------------------------- /nerf/clip_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/nerf/clip_utils.py -------------------------------------------------------------------------------- /nerf/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/nerf/datasets.py -------------------------------------------------------------------------------- /nerf/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/nerf/model_utils.py -------------------------------------------------------------------------------- /nerf/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/nerf/models.py -------------------------------------------------------------------------------- /nerf/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/nerf/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_demo.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/requirements_demo.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/run.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/codestella/putting-nerf-on-a-diet/HEAD/train.sh --------------------------------------------------------------------------------