training

TESSE Semantic Segmentation Training

Code to train semantic segmentation networks and export trained models as ONNX files.

Creating a new segmentation model

• Collect data: Lauch some TESSE environments from which to collect training data
• Train a model: Once data has been collected, run the provided notebook to train a segmentation network.

Installation

This requires Python 3.6+ with the packages listed in requirements.txt

conda create -n tesse-semantic-segmentation python=3.7 numpy scipy jupyter ipython
conda activate tesse-semantic-segmentation

pip install -r requirements.txt