├── .gitignore ├── README.md ├── conf ├── dataset │ ├── moos.yaml │ ├── pix3d.yaml │ └── scan2cad.yaml ├── default.yaml └── model │ └── cmic.yaml ├── data ├── 3dfuture │ ├── obj_dims_issue.json │ └── obj_paths_filtered.json ├── README.md ├── moos │ ├── obj_counts.json │ ├── obj_occ_rates.json │ ├── obj_views.json │ └── scene_names.txt └── shapenet │ ├── obj_dims.json │ ├── obj_paths.txt │ ├── obj_paths_filtered.json │ └── obj_scales.csv ├── docs ├── css │ └── index.css ├── dist │ ├── css │ │ ├── new_style.css │ │ └── style.css │ └── js │ │ └── main.min.js ├── images │ ├── moos_generation.png │ ├── moos_seen_unseen_qualitative.png │ ├── syn2real.png │ └── teaser.png ├── index.html └── src │ ├── js │ └── main.js │ └── scss │ ├── _normalize.scss │ ├── abstracts │ ├── _functions.scss │ ├── _include-media.scss │ ├── _mixins.scss │ └── _variables.scss │ ├── base │ ├── _base.scss │ ├── _helpers.scss │ └── _typography.scss │ ├── components │ └── _buttons.scss │ ├── layout │ ├── _header.scss │ ├── _hero.scss │ └── _main.scss │ └── style.scss ├── gcmic ├── callbacks │ ├── __init__.py │ └── cmic_callback.py ├── config │ ├── __init__.py │ └── load_cfg.py ├── dataset │ ├── future3d.py │ ├── moos.py │ ├── pix3d.py │ ├── pix3d_shape.py │ ├── sampler.py │ ├── scan2cad.py │ └── scan2cad_shape.py ├── eval │ ├── scores.py │ ├── shape_retrieval.py │ └── util.py ├── loss │ ├── __init__.py │ └── cmic_loss.py ├── model │ ├── __init__.py │ ├── cmic.py │ └── layers │ │ ├── __init__.py │ │ ├── attention.py │ │ └── encoder.py └── utils │ ├── img_util.py │ ├── io_util.py │ ├── lfd_util.py │ ├── log_util.py │ └── pt3d_util.py ├── notebook └── render_random_3dfuture_scene.ipynb ├── preprocess ├── 3dfuture │ └── get_all_lfd.py ├── moos │ ├── extract_pose_json.py │ ├── gen_dataset_hdf5.py │ ├── get_stats.py │ ├── obj_util.py │ ├── reconstruct_scene.py │ ├── reconstruct_scenes.sh │ ├── render_sample_scene.py │ ├── render_scenes.sh │ └── render_util.py ├── pix3d │ ├── extract_pose_json.py │ ├── gen_dataset_hdf5.py │ └── get_all_lfd.py └── scan2cad │ ├── extract_pose_json.py │ ├── gen_dataset_hdf5.py │ └── get_all_lfd.py ├── requirements.txt ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/README.md -------------------------------------------------------------------------------- /conf/dataset/moos.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/conf/dataset/moos.yaml -------------------------------------------------------------------------------- /conf/dataset/pix3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/conf/dataset/pix3d.yaml -------------------------------------------------------------------------------- /conf/dataset/scan2cad.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/conf/dataset/scan2cad.yaml -------------------------------------------------------------------------------- /conf/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/conf/default.yaml -------------------------------------------------------------------------------- /conf/model/cmic.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/conf/model/cmic.yaml -------------------------------------------------------------------------------- /data/3dfuture/obj_dims_issue.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/3dfuture/obj_dims_issue.json -------------------------------------------------------------------------------- /data/3dfuture/obj_paths_filtered.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/3dfuture/obj_paths_filtered.json -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/README.md -------------------------------------------------------------------------------- /data/moos/obj_counts.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/moos/obj_counts.json -------------------------------------------------------------------------------- /data/moos/obj_occ_rates.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/moos/obj_occ_rates.json -------------------------------------------------------------------------------- /data/moos/obj_views.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/moos/obj_views.json -------------------------------------------------------------------------------- /data/moos/scene_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/moos/scene_names.txt -------------------------------------------------------------------------------- /data/shapenet/obj_dims.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/shapenet/obj_dims.json -------------------------------------------------------------------------------- /data/shapenet/obj_paths.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/shapenet/obj_paths.txt -------------------------------------------------------------------------------- /data/shapenet/obj_paths_filtered.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/shapenet/obj_paths_filtered.json -------------------------------------------------------------------------------- /data/shapenet/obj_scales.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/data/shapenet/obj_scales.csv -------------------------------------------------------------------------------- /docs/css/index.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/css/index.css -------------------------------------------------------------------------------- /docs/dist/css/new_style.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/dist/css/new_style.css -------------------------------------------------------------------------------- /docs/dist/css/style.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/dist/css/style.css -------------------------------------------------------------------------------- /docs/dist/js/main.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/dist/js/main.min.js -------------------------------------------------------------------------------- /docs/images/moos_generation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/images/moos_generation.png -------------------------------------------------------------------------------- /docs/images/moos_seen_unseen_qualitative.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/images/moos_seen_unseen_qualitative.png -------------------------------------------------------------------------------- /docs/images/syn2real.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/images/syn2real.png -------------------------------------------------------------------------------- /docs/images/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/images/teaser.png -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/src/js/main.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/js/main.js -------------------------------------------------------------------------------- /docs/src/scss/_normalize.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/_normalize.scss -------------------------------------------------------------------------------- /docs/src/scss/abstracts/_functions.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/abstracts/_functions.scss -------------------------------------------------------------------------------- /docs/src/scss/abstracts/_include-media.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/abstracts/_include-media.scss -------------------------------------------------------------------------------- /docs/src/scss/abstracts/_mixins.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/abstracts/_mixins.scss -------------------------------------------------------------------------------- /docs/src/scss/abstracts/_variables.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/abstracts/_variables.scss -------------------------------------------------------------------------------- /docs/src/scss/base/_base.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/base/_base.scss -------------------------------------------------------------------------------- /docs/src/scss/base/_helpers.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/base/_helpers.scss -------------------------------------------------------------------------------- /docs/src/scss/base/_typography.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/base/_typography.scss -------------------------------------------------------------------------------- /docs/src/scss/components/_buttons.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/components/_buttons.scss -------------------------------------------------------------------------------- /docs/src/scss/layout/_header.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/layout/_header.scss -------------------------------------------------------------------------------- /docs/src/scss/layout/_hero.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/layout/_hero.scss -------------------------------------------------------------------------------- /docs/src/scss/layout/_main.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/layout/_main.scss -------------------------------------------------------------------------------- /docs/src/scss/style.scss: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/docs/src/scss/style.scss -------------------------------------------------------------------------------- /gcmic/callbacks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/callbacks/__init__.py -------------------------------------------------------------------------------- /gcmic/callbacks/cmic_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/callbacks/cmic_callback.py -------------------------------------------------------------------------------- /gcmic/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/config/__init__.py -------------------------------------------------------------------------------- /gcmic/config/load_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/config/load_cfg.py -------------------------------------------------------------------------------- /gcmic/dataset/future3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/dataset/future3d.py -------------------------------------------------------------------------------- /gcmic/dataset/moos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/dataset/moos.py -------------------------------------------------------------------------------- /gcmic/dataset/pix3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/dataset/pix3d.py -------------------------------------------------------------------------------- /gcmic/dataset/pix3d_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/dataset/pix3d_shape.py -------------------------------------------------------------------------------- /gcmic/dataset/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/dataset/sampler.py -------------------------------------------------------------------------------- /gcmic/dataset/scan2cad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/dataset/scan2cad.py -------------------------------------------------------------------------------- /gcmic/dataset/scan2cad_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/dataset/scan2cad_shape.py -------------------------------------------------------------------------------- /gcmic/eval/scores.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/eval/scores.py -------------------------------------------------------------------------------- /gcmic/eval/shape_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/eval/shape_retrieval.py -------------------------------------------------------------------------------- /gcmic/eval/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/eval/util.py -------------------------------------------------------------------------------- /gcmic/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/loss/__init__.py -------------------------------------------------------------------------------- /gcmic/loss/cmic_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/loss/cmic_loss.py -------------------------------------------------------------------------------- /gcmic/model/__init__.py: -------------------------------------------------------------------------------- 1 | from .cmic import CMIC -------------------------------------------------------------------------------- /gcmic/model/cmic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/model/cmic.py -------------------------------------------------------------------------------- /gcmic/model/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/model/layers/__init__.py -------------------------------------------------------------------------------- /gcmic/model/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/model/layers/attention.py -------------------------------------------------------------------------------- /gcmic/model/layers/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/model/layers/encoder.py -------------------------------------------------------------------------------- /gcmic/utils/img_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/utils/img_util.py -------------------------------------------------------------------------------- /gcmic/utils/io_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/utils/io_util.py -------------------------------------------------------------------------------- /gcmic/utils/lfd_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/utils/lfd_util.py -------------------------------------------------------------------------------- /gcmic/utils/log_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/utils/log_util.py -------------------------------------------------------------------------------- /gcmic/utils/pt3d_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/gcmic/utils/pt3d_util.py -------------------------------------------------------------------------------- /notebook/render_random_3dfuture_scene.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/notebook/render_random_3dfuture_scene.ipynb -------------------------------------------------------------------------------- /preprocess/3dfuture/get_all_lfd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/3dfuture/get_all_lfd.py -------------------------------------------------------------------------------- /preprocess/moos/extract_pose_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/moos/extract_pose_json.py -------------------------------------------------------------------------------- /preprocess/moos/gen_dataset_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/moos/gen_dataset_hdf5.py -------------------------------------------------------------------------------- /preprocess/moos/get_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/moos/get_stats.py -------------------------------------------------------------------------------- /preprocess/moos/obj_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/moos/obj_util.py -------------------------------------------------------------------------------- /preprocess/moos/reconstruct_scene.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/moos/reconstruct_scene.py -------------------------------------------------------------------------------- /preprocess/moos/reconstruct_scenes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/moos/reconstruct_scenes.sh -------------------------------------------------------------------------------- /preprocess/moos/render_sample_scene.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/moos/render_sample_scene.py -------------------------------------------------------------------------------- /preprocess/moos/render_scenes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/moos/render_scenes.sh -------------------------------------------------------------------------------- /preprocess/moos/render_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/moos/render_util.py -------------------------------------------------------------------------------- /preprocess/pix3d/extract_pose_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/pix3d/extract_pose_json.py -------------------------------------------------------------------------------- /preprocess/pix3d/gen_dataset_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/pix3d/gen_dataset_hdf5.py -------------------------------------------------------------------------------- /preprocess/pix3d/get_all_lfd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/pix3d/get_all_lfd.py -------------------------------------------------------------------------------- /preprocess/scan2cad/extract_pose_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/scan2cad/extract_pose_json.py -------------------------------------------------------------------------------- /preprocess/scan2cad/gen_dataset_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/scan2cad/gen_dataset_hdf5.py -------------------------------------------------------------------------------- /preprocess/scan2cad/get_all_lfd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/preprocess/scan2cad/get_all_lfd.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/generalizing_shape_retrieval/HEAD/train.py --------------------------------------------------------------------------------