├── .gitignore ├── LICENSE ├── README.md ├── config └── scannet │ ├── semaffinet_2cm.yaml │ └── semaffinet_5cm.yaml ├── dataset ├── augmentation.py ├── augmentation_2d.py ├── pregroup_2d_scannet.py ├── preprocess_3d_scannet.py ├── scanNet3D.py ├── scanNetCross.py ├── scannet │ ├── GL.py │ ├── scannet_names.txt │ ├── train_2D.txt │ └── val_2D.txt ├── voxelization_utils.py └── voxelizer.py ├── fig └── semaffinet.png ├── metrics └── iou.py ├── models ├── bpm.py ├── me_common.py ├── resnet_d.py ├── resnet_mink.py ├── semaffinet.py ├── shadownet_2d.py ├── shadownet_3d.py ├── transformer_utils │ ├── transformer.py │ └── transformer_predictor.py ├── unet_2d.py └── unet_3d.py ├── tool ├── test.py ├── test.sh ├── train.py └── train.sh └── util ├── config.py ├── criterion.py ├── solver.py └── util.py /.gitignore: -------------------------------------------------------------------------------- 1 | data 2 | Exp/ 3 | extensions/ 4 | initmodel/ 5 | __pycache__ 6 | .DS_Store -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/README.md -------------------------------------------------------------------------------- /config/scannet/semaffinet_2cm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/config/scannet/semaffinet_2cm.yaml -------------------------------------------------------------------------------- /config/scannet/semaffinet_5cm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/config/scannet/semaffinet_5cm.yaml -------------------------------------------------------------------------------- /dataset/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/augmentation.py -------------------------------------------------------------------------------- /dataset/augmentation_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/augmentation_2d.py -------------------------------------------------------------------------------- /dataset/pregroup_2d_scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/pregroup_2d_scannet.py -------------------------------------------------------------------------------- /dataset/preprocess_3d_scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/preprocess_3d_scannet.py -------------------------------------------------------------------------------- /dataset/scanNet3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/scanNet3D.py -------------------------------------------------------------------------------- /dataset/scanNetCross.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/scanNetCross.py -------------------------------------------------------------------------------- /dataset/scannet/GL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/scannet/GL.py -------------------------------------------------------------------------------- /dataset/scannet/scannet_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/scannet/scannet_names.txt -------------------------------------------------------------------------------- /dataset/scannet/train_2D.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/scannet/train_2D.txt -------------------------------------------------------------------------------- /dataset/scannet/val_2D.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/scannet/val_2D.txt -------------------------------------------------------------------------------- /dataset/voxelization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/voxelization_utils.py -------------------------------------------------------------------------------- /dataset/voxelizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/dataset/voxelizer.py -------------------------------------------------------------------------------- /fig/semaffinet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/fig/semaffinet.png -------------------------------------------------------------------------------- /metrics/iou.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/metrics/iou.py -------------------------------------------------------------------------------- /models/bpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/bpm.py -------------------------------------------------------------------------------- /models/me_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/me_common.py -------------------------------------------------------------------------------- /models/resnet_d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/resnet_d.py -------------------------------------------------------------------------------- /models/resnet_mink.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/resnet_mink.py -------------------------------------------------------------------------------- /models/semaffinet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/semaffinet.py -------------------------------------------------------------------------------- /models/shadownet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/shadownet_2d.py -------------------------------------------------------------------------------- /models/shadownet_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/shadownet_3d.py -------------------------------------------------------------------------------- /models/transformer_utils/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/transformer_utils/transformer.py -------------------------------------------------------------------------------- /models/transformer_utils/transformer_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/transformer_utils/transformer_predictor.py -------------------------------------------------------------------------------- /models/unet_2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/unet_2d.py -------------------------------------------------------------------------------- /models/unet_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/models/unet_3d.py -------------------------------------------------------------------------------- /tool/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/tool/test.py -------------------------------------------------------------------------------- /tool/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/tool/test.sh -------------------------------------------------------------------------------- /tool/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/tool/train.py -------------------------------------------------------------------------------- /tool/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/tool/train.sh -------------------------------------------------------------------------------- /util/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/util/config.py -------------------------------------------------------------------------------- /util/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/util/criterion.py -------------------------------------------------------------------------------- /util/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/util/solver.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangzy22/SemAffiNet/HEAD/util/util.py --------------------------------------------------------------------------------