Clone of VLSAT repository to work with different others dataset.
-
Follow instruction for installation of original VLSAT: https://github.com/wz7in/CVPR2023-VLSAT.git
-
Generate data with ScanNet dataset. Go to data_processing/:
For GT ScanNet segmentation:
python gen_data_scannet_gt.pyFor Mask3D ScanNet segmentation (use Chat-Scene segmentation files):
python gen_data_scannet_mask3d.pyFor OneFormer3D ScanNet segmentation:
python gen_data_scannet_oneformer3d.py- Run patched evaluation with pre-trained VLSAT weights 3dssg_best_ckpt (available on https://github.com/wz7in/CVPR2023-VLSAT.git)
python -m main --mode eval --config config/mmgnet.json --exp /home/jovyan/Tatiana_Z/3dssg_best_ckpt- Run prepare_filtered_gnn_data.py (choose graph hyperparameters at the beginning of the script)
python prepare_filtered_gnn_data.py