文件名称:setvae:论文“ SetVAE的官方资料库
文件大小:4.72MB
文件格式:ZIP
更新时间:2024-04-21 09:29:54
point-cloud pytorch transformer vae attention-mechanism
SetVAE-官方PyTorch实施 该存储库是本文的正式实现: 。 , , , (*等额捐款) CVPR 2021 安装 您可以使用Docker映像wogns98/setvae ,也可以使用提供的Dockerfile构建自己的Docker映像。 docker pull wogns98/setvae:latest 如果您不想使用Docker,请按照以下步骤进行; 但是我们强烈建议您使用Docker。 sudo apt-get update sudo apt-get install python3.6 git clone https://github.com/jw9730/setvae.git setvae cd setvae pip install -r requirements.txt bash install.sh 数据集 对于MNIST和Set-MultiMNIST,数
【文件预览】:
setvae-master
----args.py(12KB)
----.gitignore(110B)
----Dockerfile(242B)
----install.sh(146B)
----sample_and_summarize.py(6KB)
----batch_size.json(28B)
----requirements.txt(377B)
----datasets()
--------ShapeNet.py(13KB)
--------__init__.py(342B)
--------MultiMNIST.py(8KB)
--------SetMNIST.py(7KB)
--------SetMultiMNIST.py(7KB)
----models()
--------mlp.py(2KB)
--------layers.py(11KB)
--------fiblat()
--------criterion.py(4KB)
--------networks.py(14KB)
--------ops.py(6KB)
----engine.py(5KB)
----utils.py(28KB)
----scripts()
--------shapenet_airplane.sh(1KB)
--------shapenet_airplane_test.sh(779B)
--------shapenet_chair_test.sh(773B)
--------shapenet_car_test.sh(769B)
--------mnist.sh(942B)
--------shapenet_chair.sh(1KB)
--------shapenet_car_viz.sh(770B)
--------multimnist.sh(1KB)
--------shapenet_airplane_viz.sh(762B)
--------shapenet_chair_viz.sh(756B)
--------mnist_viz.sh(617B)
--------multimnist_viz.sh(693B)
--------shapenet_car.sh(1KB)
----README.md(5KB)
----figures()
--------MNIST_viz_attn.ipynb(7KB)
--------open3d_viz_samples.ipynb(8KB)
--------open3d_viz_attn_car.ipynb(12KB)
--------visualizer.py(3KB)
--------MNIST_viz_samples.ipynb(6KB)
--------camera_config.json(658B)
--------open3d_viz_attn_airplane.ipynb(12KB)
--------open3d_viz_attn_chair.ipynb(12KB)
--------utils.py(2KB)
--------cardinality_generalization.ipynb(9KB)
--------colormap.py(3KB)
--------draw.py(17KB)
----sample_and_test_all.py(5KB)
----batch_size_128.json(29B)
----docs()
--------github_key.png(4.7MB)
----metrics()
--------.gitignore(62B)
--------__init__.py(168B)
--------utils.py(271B)
--------evaluation_metrics.py(11KB)
--------pytorch_structural_losses()
--------losses.py(3KB)
----train.py(8KB)