Code for "FPS-Net: A convolutional fusion network for large-scale LiDAR point cloud segmentation".

Overview

arXiv

FPS-Net

Code for "FPS-Net: A convolutional fusion network for large-scale LiDAR point cloud segmentation", accepted by ISPRS journal of Photogrammetry and Remote Sensing
By Aoran Xiao, Xiaofei Yang, Shijian Lu, Dayan Guan, Jiaxing Huang

Full Paper

Install

conda create -n FPSNet python=3.7
source activate FPSNet
cd /ROOT/
pip install -r requirements.txt

Train

cd /train/tasks/semantic
sh train.sh

Inference and Test

cd /train/tasks/semantic
sh test.sh

Citation

If you use this code, please cite:

@article{xiao2021fps,
  title={FPS-Net: A convolutional fusion network for large-scale LiDAR point cloud segmentation},
  author={Xiao, Aoran and Yang, Xiaofei and Lu, Shijian and Guan, Dayan and Huang, Jiaxing},
  journal={ISPRS Journal of Photogrammetry and Remote Sensing},
  volume={176},
  pages={237--249},
  year={2021},
  publisher={Elsevier}
}

Acknowledgement

Code borrowed heavily from lidar-bonnetal, thanks for their sharing!

Comments
  • What is the structure of the dataset?

    What is the structure of the dataset?

    After I downloaded the code from the website, it did get an error. The reason is that the sequences folder under my data folder is empty. I didn't see the relevant content of the dataset from the README. I want to know that this error is due to me Is it caused by not preparing the data set, and what are the requirements for the data set of the code?

    opened by yuyunlong2002 4
  • A question about depth and xyz

    A question about depth and xyz

    Hello, I want to ask why we should use depth and xyz at the same time. I observed in your code that the depth is calculated using xyz(depth=np.linalg.norm(self.points, 2, axis=1)), so there is a correlation between the two. Why do we use depth when using xyz , is there any result indicating that using depth and xyz simultaneously will have better effect?

    opened by yuyunlong2002 2
  • A bug occurred when running infer.py in test.sh

    A bug occurred when running infer.py in test.sh

    Sorry to bother you again,When I running your input in test.sh an error occurred in the. py file. There are four variables assigned to a line of code (including proj_output, range_feat, zxy_feat, and resission_feat), but the actual return value of the function is only one (y). I see in your function body that a line of commented content returns four values. How can I solve this problem? code:https://github.com/xiaoaoran/FPS-Net/blob/main/train/tasks/semantic/modules/user.py#L138 image

    Function body return:https://github.com/xiaoaoran/FPS-Net/blob/main/train/tasks/semantic/modules/segmentator.py#L166 image

    error: image

    opened by yuyunlong2002 2
  • A problem about the projection of heterogeneous modal graphs in networks

    A problem about the projection of heterogeneous modal graphs in networks

    Hello, I would like to ask where is the content in the code about converting 3D data into 2D images of different modes you said in your paper,In the FPS net diagram you built, the following parts are shown: 1

    opened by yuyunlong2002 2
  • A question about the SemanticKITTI dataset

    A question about the SemanticKITTI dataset

    Hello,sry to bother you again, but as a KITTI beginner, a question bother me for a long time. Is the coordinate system of KITTI point cloud data based on the 3D laser scanner? If so, shouldn't the origin coordinates of a frame of point cloud change instantaneously when the car is moving (the 3D laser scanner will move with the car moving at every moment), because the scanning frequency is too fast and KITTI data ignores this local coordinate system?

    opened by yuyunlong2002 1
Owner
Ph.D. student
null
TensorFlow code for the neural network presented in the paper: "Structural Language Models of Code" (ICML'2020)

SLM: Structural Language Models of Code This is an official implementation of the model described in: "Structural Language Models of Code" [PDF] To ap

null 73 Nov 6, 2022
Inference code for "StylePeople: A Generative Model of Fullbody Human Avatars" paper. This code is for the part of the paper describing video-based avatars.

NeuralTextures This is repository with inference code for paper "StylePeople: A Generative Model of Fullbody Human Avatars" (CVPR21). This code is for

Visual Understanding Lab @ Samsung AI Center Moscow 18 Oct 6, 2022
A code generator from ONNX to PyTorch code

onnx-pytorch Generating pytorch code from ONNX. Currently support onnx==1.9.0 and torch==1.8.1. Installation From PyPI pip install onnx-pytorch From

Wenhao Hu 94 Jan 6, 2023
This is the code for our KILT leaderboard submission to the T-REx and zsRE tasks. It includes code for training a DPR model then continuing training with RAG.

KGI (Knowledge Graph Induction) for slot filling This is the code for our KILT leaderboard submission to the T-REx and zsRE tasks. It includes code fo

International Business Machines 72 Jan 6, 2023
Convert Python 3 code to CUDA code.

Py2CUDA Convert python code to CUDA. Usage To convert a python file say named py_file.py to CUDA, run python generate_cuda.py --file py_file.py --arch

Yuval Rosen 3 Jul 14, 2021
Empirical Study of Transformers for Source Code & A Simple Approach for Handling Out-of-Vocabulary Identifiers in Deep Learning for Source Code

Transformers for variable misuse, function naming and code completion tasks The official PyTorch implementation of: Empirical Study of Transformers fo

Bayesian Methods Research Group 56 Nov 15, 2022
Reference implementation of code generation projects from Facebook AI Research. General toolkit to apply machine learning to code, from dataset creation to model training and evaluation. Comes with pretrained models.

This repository is a toolkit to do machine learning for programming languages. It implements tokenization, dataset preprocessing, model training and m

Facebook Research 408 Jan 1, 2023
Code for the prototype tool in our paper "CoProtector: Protect Open-Source Code against Unauthorized Training Usage with Data Poisoning".

CoProtector Code for the prototype tool in our paper "CoProtector: Protect Open-Source Code against Unauthorized Training Usage with Data Poisoning".

Zhensu Sun 1 Oct 26, 2021
Low-code/No-code approach for deep learning inference on devices

EzEdgeAI A concept project that uses a low-code/no-code approach to implement deep learning inference on devices. It provides a componentized framewor

On-Device AI Co., Ltd. 7 Apr 5, 2022
Code for all the Advent of Code'21 challenges mostly written in python

Advent of Code 21 Code for all the Advent of Code'21 challenges mostly written in python. They are not necessarily the best or fastest solutions but j

null 4 May 26, 2022
Code to use Augmented Shapiro Wilks Stopping, as well as code for the paper "Statistically Signifigant Stopping of Neural Network Training"

This codebase is being actively maintained, please create and issue if you have issues using it Basics All data files are included under losses and ea

J K Terry 32 Nov 9, 2021
Opinionated code formatter, just like Python's black code formatter but for Beancount

beancount-black Opinionated code formatter, just like Python's black code formatter but for Beancount Try it out online here Features MIT licensed - b

Launch Platform 16 Oct 11, 2022
a delightful machine learning tool that allows you to train, test and use models without writing code

igel A delightful machine learning tool that allows you to train/fit, test and use models without writing code Note I'm also working on a GUI desktop

Nidhal Baccouri 3k Jan 5, 2023
Pytorch Lightning code guideline for conferences

Deep learning project seed Use this seed to start new deep learning / ML projects. Built in setup.py Built in requirements Examples with MNIST Badges

Pytorch Lightning 1k Jan 2, 2023
Automatically Build Multiple ML Models with a Single Line of Code. Created by Ram Seshadri. Collaborators Welcome. Permission Granted upon Request.

Auto-ViML Automatically Build Variant Interpretable ML models fast! Auto_ViML is pronounced "auto vimal" (autovimal logo created by Sanket Ghanmare) N

AutoViz and Auto_ViML 397 Dec 30, 2022
Code samples for my book "Neural Networks and Deep Learning"

Code samples for "Neural Networks and Deep Learning" This repository contains code samples for my book on "Neural Networks and Deep Learning". The cod

Michael Nielsen 13.9k Dec 26, 2022
Code for: https://berkeleyautomation.github.io/bags/

DeformableRavens Code for the paper Learning to Rearrange Deformable Cables, Fabrics, and Bags with Goal-Conditioned Transporter Networks. Here is the

Daniel Seita 121 Dec 30, 2022
Code for our method RePRI for Few-Shot Segmentation. Paper at http://arxiv.org/abs/2012.06166

Region Proportion Regularized Inference (RePRI) for Few-Shot Segmentation In this repo, we provide the code for our paper : "Few-Shot Segmentation Wit

Malik Boudiaf 138 Dec 12, 2022
Applications using the GTN library and code to reproduce experiments in "Differentiable Weighted Finite-State Transducers"

gtn_applications An applications library using GTN. Current examples include: Offline handwriting recognition Automatic speech recognition Installing

Facebook Research 68 Dec 29, 2022