├── .gitignore ├── LICENSE ├── README.md ├── calculate_f1.py ├── config ├── config.yaml └── data │ ├── base.yaml │ ├── text2shape_c13.yaml │ └── text2shape_chair_table.yaml ├── data └── preprocess_all_data.py ├── docs ├── bootstrap.min.css ├── img │ ├── overview.png │ ├── paper_thumbnails.png │ └── teaser.png ├── index.html └── offcanvas.css ├── eval.py ├── extract_clip_feats.py ├── setup.py ├── test.py ├── train.py └── tricolo ├── callback └── lr_decay_callback.py ├── data ├── data_module.py └── dataset │ ├── __init__.py │ ├── general_dataset.py │ ├── text2shape_c13.py │ └── text2shape_chair_table.py ├── evaluation └── eval_retrieval.py ├── loss ├── nt_xent.py └── triplet.py └── model ├── module ├── img_encoder │ ├── clip_img.py │ └── mv_cnn.py ├── text_encoder │ ├── bigru.py │ └── clip_text.py └── voxel_encoder │ └── sparse_cnn.py └── tricolo_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/README.md -------------------------------------------------------------------------------- /calculate_f1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/calculate_f1.py -------------------------------------------------------------------------------- /config/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/config/config.yaml -------------------------------------------------------------------------------- /config/data/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/config/data/base.yaml -------------------------------------------------------------------------------- /config/data/text2shape_c13.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/config/data/text2shape_c13.yaml -------------------------------------------------------------------------------- /config/data/text2shape_chair_table.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/config/data/text2shape_chair_table.yaml -------------------------------------------------------------------------------- /data/preprocess_all_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/data/preprocess_all_data.py -------------------------------------------------------------------------------- /docs/bootstrap.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/docs/bootstrap.min.css -------------------------------------------------------------------------------- /docs/img/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/docs/img/overview.png -------------------------------------------------------------------------------- /docs/img/paper_thumbnails.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/docs/img/paper_thumbnails.png -------------------------------------------------------------------------------- /docs/img/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/docs/img/teaser.png -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/offcanvas.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/docs/offcanvas.css -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/eval.py -------------------------------------------------------------------------------- /extract_clip_feats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/extract_clip_feats.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/setup.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/train.py -------------------------------------------------------------------------------- /tricolo/callback/lr_decay_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/callback/lr_decay_callback.py -------------------------------------------------------------------------------- /tricolo/data/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/data/data_module.py -------------------------------------------------------------------------------- /tricolo/data/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/data/dataset/__init__.py -------------------------------------------------------------------------------- /tricolo/data/dataset/general_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/data/dataset/general_dataset.py -------------------------------------------------------------------------------- /tricolo/data/dataset/text2shape_c13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/data/dataset/text2shape_c13.py -------------------------------------------------------------------------------- /tricolo/data/dataset/text2shape_chair_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/data/dataset/text2shape_chair_table.py -------------------------------------------------------------------------------- /tricolo/evaluation/eval_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/evaluation/eval_retrieval.py -------------------------------------------------------------------------------- /tricolo/loss/nt_xent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/loss/nt_xent.py -------------------------------------------------------------------------------- /tricolo/loss/triplet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/loss/triplet.py -------------------------------------------------------------------------------- /tricolo/model/module/img_encoder/clip_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/model/module/img_encoder/clip_img.py -------------------------------------------------------------------------------- /tricolo/model/module/img_encoder/mv_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/model/module/img_encoder/mv_cnn.py -------------------------------------------------------------------------------- /tricolo/model/module/text_encoder/bigru.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/model/module/text_encoder/bigru.py -------------------------------------------------------------------------------- /tricolo/model/module/text_encoder/clip_text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/model/module/text_encoder/clip_text.py -------------------------------------------------------------------------------- /tricolo/model/module/voxel_encoder/sparse_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/model/module/voxel_encoder/sparse_cnn.py -------------------------------------------------------------------------------- /tricolo/model/tricolo_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/3dlg-hcvc/tricolo/HEAD/tricolo/model/tricolo_net.py --------------------------------------------------------------------------------