Learning a deep neural network for ground segmentation.
git clone https://github.com/gaunthan/ground-segmentation.git
cd ground-segmentation
pip install -r requirements.txt
Run get_dataset.py to download the dataset. We use cmu-corridor-dataset to train the model.
Under ground-segmentation, run the following command to launch JupyterLab
jupyter-lab .
Model definition, training and inference are written in notebook ground_segmentation.ipynb.
you can open it in JupyterLab.
Or, you can view the notebook on web browser by Jupyter Notebook Viewer.
Click the lick to view.