Source code for the paper:
Jingwei Huang, Yanfeng Zhang, Mingwei Sun. [PrimitiveNet: Primitive Instance Segmentation with Local Primitive Embedding under Adversarial Metric], ICCV 2021 .
git submodule update --init --recursive sh scripts/compile.sh
Download data and checkpoints
Follow the comments in download.sh to download the data and checkpoints.
Evaluation on ABC
Predicted results on test set are visualized in src/results/visualize. Original network predictions are saved in src/results/predictions.
After all test set predictions are generated, an evaluation for mSegIOU/mLabelIOU/APs will be executed and final results will be saved at src/results/statistics.
Train ABC from scratch
Logs and trained models will be saved at src/results/checkpoint.
Execute on a large scene
The segmented large scene is stored at src/results/visualize/final.obj.