bharathrajm/fpcc

FPCC: Fast Point Cloud Clustering-based Instance Segmentation for Industrial Bin-picking [Arxiv]

Other Implementation

• pytorch-fpcc

Citation

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

@article{XU2022255,
title = {FPCC: Fast point cloud clustering-based instance segmentation for industrial bin-picking},
journal = {Neurocomputing},
volume = {494},
pages = {255-268},
year = {2022},
issn = {0925-2312},
doi = {https://doi.org/10.1016/j.neucom.2022.04.023},
url = {https://www.sciencedirect.com/science/article/pii/S0925231222003915},
author = {Yajun Xu and Shogo Arai and Diyi Liu and Fangzhou Lin and Kazuhiro Kosuge},
keywords = {Bin-picking, 3D Point Cloud, Instance segmentaion, Deep Learning},
}

Dependencies

• tensorflow (1.13.1)
• h5py

Data generation

Thanks for [waiyc] for providing a [script] to generate synthetic data by Pybullet. The dataset is generate and recorded base on the steps mentioned in IPA Dataset.

Training & Testing

python fpcc_train.py 

python fpcc_test.py

Evaluation metric

We use the code provided by [ASIS] to calculate precision and recall.

XA Dataset

XA dataset can be downloaded [here]. Please cite this paper or "FPCC" if you want to use XA dataset in your work,

@ARTICLE{9025047,
 author={Xu, Yajun and Arai, Shogo and Tokuda, Fuyuki and Kosuge, Kazuhiro},
 journal={IEEE Access},
 title={A Convolutional Neural Network for Point Cloud Instance Segmentation in Cluttered Scene Trained by Synthetic Data Without Color},
 year={2020},
 volume={8},
 number={},
 pages={70262-70269},
 doi={10.1109/ACCESS.2020.2978506}
}

IPA Dataset

The information about IPA dataset can be found [here]. The paper about IPA dataset can be downloaded [here]. The author of IPA did not provide a public link to download the data set, so maybe you need to register first.

We only uploaded part of the IPA dataset in the "datas" folder. Use the following scripts for generating h5 files for traing.

python convert_csv2json_annotation.py
python IPA_image2pc.py
python generate_ipa_center_h5.py
python generate_file_list.py

Acknowledgemets

This project is built upon [PointNet], [PointNet++], [SGPN] and [DGCNN]

Others

The program is not very beautiful. If you have any questions, please feel free to contact me at the address below. [email protected]