Skip to content

jimmy15923/wspss_mil_transformer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

An MIL-Derived Transformer for Weakly SupervisedPoint Cloud Segmentation

This repository is the implementation of An MIL-Derived Transformer for Weakly SupervisedPoint Cloud Segmentation.

Requirements

  • Install MinkowskiEngine by using Conda

Data Preparation

  • Download the dataset from the official ScanNet and S3DIS websites.

Training

  • Start training: sh tool/train.sh EXP_NAME /PATH/TO/CONFIG NUMBER_OF_THREADS

  • Resume: sh tool/resume.sh EXP_NAME /PATH/TO/CONFIG(copied one) NUMBER_OF_THREADS

NUMBER_OF_THREADS is the threads to use per process (gpu), so optimally, it should be Total_threads / gpu_number_used

Citation

If you find our work useful in your research, please consider citing our paper:

@inproceedings{yang2022mil,
  title={An mil-derived transformer for weakly supervised point cloud segmentation},
  author={Yang, Cheng-Kun and Wu, Ji-Jia and Chen, Kai-Syun and Chuang, Yung-Yu and Lin, Yen-Yu},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={11830--11839},
  year={2022}
}

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published